Update logic.py

logic.py

@@ -280,11 +280,43 @@ def login_ArangoDB():
     print("Database: " + login["dbName"])
     return login
 
+def create_smart_graph():
+  # defining graph schema
+
+  # create a new graph called movie_rating_graph in the temp database if it does not already exist.
+  if not movie_rec_db.has_graph("movie_rating_graph"):
+      movie_rec_db.create_graph('movie_rating_graph', smart=True)
+
+  # This returns and API wrapper for the above created graphs
+  movie_rating_graph = movie_rec_db.graph("movie_rating_graph")
+
+  # Create a new vertex collection named "Users" if it does not exist.
+  if not movie_rating_graph.has_vertex_collection("Users"):
+      movie_rating_graph.vertex_collection("Users")
+
+  # Create a new vertex collection named "Movie" if it does not exist.
+  if not movie_rating_graph.has_vertex_collection("Movie"):
+      movie_rating_graph.vertex_collection("Movie")
+
+  # creating edge definitions named "Ratings. This creates any missing
+  # collections and returns an API wrapper for "Ratings" edge collection.
+  if not movie_rating_graph.has_edge_definition("Ratings"):
+      Ratings = movie_rating_graph.create_edge_definition(
+          edge_collection='Ratings',
+          from_vertex_collections=['Users'],
+          to_vertex_collections=['Movie']
+      )
+
+  return movie_rating_graph, Ratings
+
 def load_data_to_ArangoDB(login):
-    
+    global movie_rec_db
     movie_rec_db = oasis.connect_python_arango(login)
     if not movie_rec_db.has_collection("Movie"):
         movie_rec_db.create_collection("Movie", replication_factor=3)
+
+
+    movie_rating_graph, Ratings = create_smart_graph()
         
     batch = []
     BATCH_SIZE = 128
@@ -352,27 +384,6 @@ def load_data_to_ArangoDB(login):
     if not movie_rec_db.has_collection("Ratings"):
         movie_rec_db.create_collection("Ratings", edge=True, replication_factor=3)
 
-    if not movie_rec_db.has_graph("movie_rating_graph"):
-        movie_rec_db.create_graph('movie_rating_graph', smart=True)
-    
-    # This returns and API wrapper for the above created graphs
-    movie_rating_graph = movie_rec_db.graph("movie_rating_graph")
-
-    # Create a new vertex collection named "Users" if it does not exist.
-    if not movie_rating_graph.has_vertex_collection("Users"):
-        movie_rating_graph.vertex_collection("Users")
-
-    # Create a new vertex collection named "Movie" if it does not exist.
-    if not movie_rating_graph.has_vertex_collection("Movie"):
-        movie_rating_graph.vertex_collection("Movie")
-
-    if not movie_rating_graph.has_edge_definition("Ratings"):
-        Ratings = movie_rating_graph.create_edge_definition(
-            edge_collection='Ratings',
-            from_vertex_collections=['Users'],
-            to_vertex_collections=['Movie']
-        )
-
     user_id, movie_id, ratings = ratings_df[['userId']].values.flatten(), ratings_df[['movieId']].values.flatten() , ratings_df[['rating']].values.flatten()
     create_ratings_graph(user_id, movie_id, ratings)