@@ -89,6 +89,7 @@ def save_to_database(
8989 connector : ResourceConnector ,
9090 router : ResourceRouter ,
9191 item : RESOURCE | ResourceWithRelations [RESOURCE ] | RecordError ,
92+ user ,
9293) -> Optional [RecordError ]:
9394 if isinstance (item , RecordError ):
9495 return item
@@ -103,7 +104,7 @@ def save_to_database(
103104 )
104105 # TODO: if not None, update (https://github.com/aiondemand/AIOD-rest-api/issues/131)
105106 if existing is None :
106- resource = router .create_resource (session , resource_create_instance )
107+ resource = router .create_resource (session , resource_create_instance , user )
107108 publish_resource (session , resource )
108109
109110 except Exception as e :
@@ -217,9 +218,18 @@ def main():
217218 if router .resource_class == connector .resource_class
218219 ]
219220
221+ class ConnectorUser :
222+ def is_connector (self ) -> bool :
223+ return True
224+
225+ def is_connector_for_platform (self , platform : str ) -> bool :
226+ return platform == connector .platform_name
227+
220228 with DbSession () as session :
221229 for i , item in enumerate (items ):
222- error = save_to_database (router = router , connector = connector , session = session , item = item )
230+ error = save_to_database (
231+ router = router , connector = connector , session = session , item = item , user = ConnectorUser ()
232+ )
223233 if error :
224234 if not error .ignore :
225235 if isinstance (error .error , str ):
0 commit comments