diff --git a/driver.py b/driver.py index 928d4b4..d0f49e2 100644 --- a/driver.py +++ b/driver.py @@ -17,26 +17,18 @@ async def main(): ffrom = urls[0] to= urls[0] - from_id = "" - to_id = "" + to_id = None + from_id = None + + print(ffrom, "->", to) # FROM result = await db.query(f"SELECT id FROM website WHERE url='{ffrom}'") if (len(result[0]['result']) > 0): - # The record exists - await db.update('website', { - 'url':ffrom, - 'crawled':True, - }) from_id = result[0]['result'][0]['id'] + await db.query(f"UPDATE {from_id} MERGE {{ crawled: True }}") else: - from_response = await db.create( - "website", - { - "url":ffrom, - "crawled":True, - }, - ) + from_response = await db.create("website", { "url":ffrom, "crawled":True, } ) from_id = from_response[0]["id"] # TO @@ -48,9 +40,6 @@ async def main(): else: to_id = result[0]['result'][0]['id'] - print("From: ", from_id) - print("To: ", to_id) - await db.query(f"RELATE {from_id} -> links_to -> {to_id}"); if __name__ == "__main__":