new commit

98__r__importArtifactToMaterialRelation.py

@@ -5,79 +5,66 @@ import os # For environment variable loading
 from dotenv import load_dotenv # For environment variable loading
 import pandas as pd # For dataframe handling
 
-# Initialize the database
-print('Initializing the database...')
-engine, metadata, Session = initDb(True, './schemas/')
-if engine == False:
-    print('Database initialization failed.')
-    exit()
+def importArtifactToMaterialRelation(api, engine):
+    print('importing artifact to material relation')
 
-# Load the environment variables
-load_dotenv()
+    tableName = "r__obj__5280_material"
+    bundleId = 'b825aff7df3d48bd875e2a081c796305' # Artifact to material relation
+    try:
+        processedRows = pd.read_csv(f'./logs/{tableName}.csv')
+    except FileNotFoundError:
+        processedRows = pd.DataFrame(columns=[ 'id', 'uuid', 'uri'])
 
-# Initialize the WissKI API
-print('Initializing the WissKI API...')
-api_url = os.getenv('API_URL')
-auth = (os.getenv('API_USERNAME'), os.getenv('API_PASSWORD'))
-headers = {"Cache-Control": "no-cache"}
-api = Api(api_url, auth, headers)
-api.pathbuilder = api.get_pathbuilder('relations')
+    # Load sources table
+    sqlTable = pd.read_sql_table(tableName, con=engine)
 
-
-tableName = "r__obj__5280_material"
-bundleId = 'b825aff7df3d48bd875e2a081c796305' # Artifact to material relation
-try:
-    processedRows = pd.read_csv(f'./logs/processed-{tableName}.csv')
-except FileNotFoundError:
-    processedRows = pd.DataFrame(columns=['docId', 'uuid', 'uri'])
-
-# Load sources table
-sqlTable = pd.read_sql_table(tableName, con=engine)
-
-entityValues = {}
-
-# Create entities
-for index, row in sqlTable.iterrows():
-    # For every row in table...
-    if index < len(processedRows) and sqlTable.loc[index, 'id'] == processedRows.loc[index, 'docId']:
-        # skip if already processed
-        print(f'Skipping already processed entity {sqlTable.loc[index, 'id']}')
-        continue
-    # Create Entity property dicts
     entityValues = {}
-    for key, value in row.items():
-        # For every column in row...
-        if (value is None) or (value == ''):
-            #  skip if cell has no value
+
+    # Create entities
+    for index, row in sqlTable.iterrows():
+        # For every row in table...
+        if index < len(processedRows) and sqlTable.loc[index, 'id'] == processedRows.loc[index, 'id']:
+            # skip if already processed
+            print(f'Skipping already processed entity {sqlTable.loc[index, "id"]}')
             continue
-        # Properties of an entity have to be an array, so...
-        if '&' in str(value):
-            # ...Explode "&"-separated values to array items
-            value = [x.strip() for x in str(value).split('&')]
-        else:
-            # ...Or parse to array
-            value = [value]
-        # Map columns to fields. We use assignments for reification.
-        docId = ''
-        match key:
-            case 'id':
-                docId = value[0]
-            case 'f__obj__uuid':
-                entityValues['f9f07bf63ccafd4eb2c0de24c73e1664'] = value # Artifact UUID
-                fUuid = value[0]
-            case 'f__5280_material__uuid':
-                entityValues['f820534abde4c2a2d19e0d19f7793cf0'] = value #
-            case _:
-                print(f'{key} is not a valid field, skipping.')
+        # Create Entity property dicts
+        entityValues = {}
+        for key, value in row.items():
+            # For every column in row...
+            if (value is None) or (value == ''):
+                #  skip if cell has no value
+                continue
+            # Properties of an entity have to be an array, so...
+            value = str(value).replace('&###{{new_line}}###'.format(), '&')
+            value = str(value).replace('###{{new_line}}###', '&')
+            value = str(value).replace(' & ', '&')
+            if '&' in str(value):
+                # ...Explode "&"-separated values to array items
+                value = [x.strip() for x in str(value).split('&')]
+            else:
+                # ...Or parse to array
+                value = [value]
+            # Map columns to fields. We use assignments for reification.
+            docId = ''
+            match key:
+                case 'id':
+                    docId = value[0]
+                case 'f__obj__uuid':
+                    entityValues['f9f07bf63ccafd4eb2c0de24c73e1664'] = value # Artifact UUID
+                    fUuid = value[0]
+                case 'f__5280_material__uuid':
+                    entityValues['f820534abde4c2a2d19e0d19f7793cf0'] = value #
+                case _:
+                    print(f'{key} is not a valid field, skipping.')
 
-    # Create Material
-    entity = Entity(api=api, fields=entityValues, bundle_id=bundleId)
-    api.save(entity)
+        # Create Material
+        entity = Entity(api=api, fields=entityValues, bundle_id=bundleId)
+        api.save(entity)
 
-    print(f'Created entity {index}: {entity.uri} of {len(sqlTable)}')
+        print(f'Created Artifact to Material Relation {index}: {entity.uri} of {len(sqlTable)}')
 
-    # Write log
-    processedRows = processedRows._append({'docId': docId, 'uuid': fUuid, 'uri': entity.uri}, ignore_index=True)
-    processedRows.to_csv(f'./logs/processed-{tableName}.csv', index=False)
+        # Write log
+        processedRows = processedRows._append({'id': row['id'], 'uuid': fUuid, 'uri': entity.uri}, ignore_index=True)
+        processedRows.to_csv(f'./logs/{tableName}.csv', index=False)
 
-print('finish')
+    print('finished importing artifact to material relation')