Fix up linting issues

Author: JackWilb

multinet/api/management/commands/setupdevenv.py

@@ -1,68 +1,60 @@
-from arango.exceptions import (
-    ArangoClientError,
-    ArangoServerError,
-    PermissionUpdateError,
-    UserCreateError,
-)
+import csv
+from io import StringIO
+import json
+import os
+from pathlib import Path
+
 from django.conf import settings
-from django.core.management.base import BaseCommand
 from django.contrib.auth.models import User
+from django.core.management.base import BaseCommand
 
-from multinet.api.utils.arango import arango_system_db
-from multinet.api.models import Workspace, Network, Table, TableTypeAnnotation
+from multinet.api.models import Network, Table, TableTypeAnnotation, Workspace
 from multinet.api.tasks.upload.csv import process_row
 
-from pathlib import Path
-from io import StringIO
-import os
-import csv
-import json
-
-
 
 class Command(BaseCommand):
-    help = "Sets up a standard development environment with example workspaces and tables."
+    help = 'Sets up a standard development environment with example workspaces and tables.'
 
     def create_workspace_if_not_exists(self, workspace_name):
         try:
             # Check if workspace exists, if so remove it
             if Workspace.objects.filter(name=workspace_name).exists():
-                self.stdout.write(self.style.SUCCESS(
-                    f"{workspace_name} already exists, removing to recreate"
-                ))
+                self.stdout.write(
+                    self.style.SUCCESS(f'{workspace_name} already exists, removing to recreate')
+                )
                 Workspace.objects.filter(name=workspace_name).delete()
 
             # Create the workspace
             Workspace.objects.create(name=workspace_name, public=True, owner=self.user)
-            self.stdout.write(self.style.SUCCESS(
-                f"{workspace_name} created"
-            ))
+            self.stdout.write(self.style.SUCCESS(f'{workspace_name} created'))
 
         except Exception as e:
             self.stderr.write(self.style.ERROR(str(e)))
 
     def create_tables_for_workspace(self, workspace_name, edge_table_name):
         try:
             # Get the paths for all the data objects to upload
-            data_dir_path = Path(settings.BASE_DIR) / "data" / workspace_name
-            csv_paths = list(data_dir_path.glob("*.csv"))
+            data_dir_path = Path(settings.BASE_DIR) / 'data' / workspace_name
+            csv_paths = list(data_dir_path.glob('*.csv'))
             workspace = Workspace.objects.get(name=workspace_name)
 
             for csv_path in csv_paths:
                 filename = os.path.splitext(csv_path.name)[0]
-                
+
                 # Check if table with name exists, if so delete
                 if Table.objects.filter(workspace=workspace, name=filename).exists():
-                    self.stdout.write(self.style.SUCCESS(
-                        f"{workspace_name}/{filename} already exists, removing to recreate"
-                    ))
+                    self.stdout.write(
+                        self.style.SUCCESS(
+                            f'{workspace_name}/{filename} already exists, removing to recreate'
+                        )
+                    )
                     Table.objects.filter(workspace=workspace, name=filename).delete()
 
                 # Open the types file and read the types in
-                types_path = data_dir_path / "types" / f"{filename}.json"
+                types_path = data_dir_path / 'types' / f'{filename}.json'
                 with types_path.open('rb') as f:
                     columns = json.load(f)
-                    
+
                 # Open the csv and read in the rows
                 with csv_path.open('rb') as f:
                     csv_rows = list(csv.DictReader(StringIO(f.read().decode('utf-8'))))
@@ -72,7 +64,9 @@ def create_tables_for_workspace(self, workspace_name, edge_table_name):
                     csv_rows[i] = process_row(row, columns)
 
                 # Create the table
-                new_table = Table.objects.create(workspace=workspace, name=filename, edge=filename==edge_table_name)
+                new_table = Table.objects.create(
+                    workspace=workspace, name=filename, edge=filename == edge_table_name
+                )
 
                 # Put the rows into the table
                 new_table.put_rows(csv_rows)
@@ -85,9 +79,7 @@ def create_tables_for_workspace(self, workspace_name, edge_table_name):
                     ]
                 )
 
-                self.stdout.write(self.style.SUCCESS(
-                    f"{workspace_name}/{filename} created"
-                ))
+                self.stdout.write(self.style.SUCCESS(f'{workspace_name}/{filename} created'))
 
         except Exception as e:
             self.stderr.write(self.style.ERROR(str(e)))
@@ -97,10 +89,10 @@ def create_network_in_workspace(self, workspace_name, edge_table_name, node_tabl
             workspace = Workspace.objects.get(name=workspace_name)
 
             # Create the network
-            Network.create_with_edge_definition(workspace_name, workspace, edge_table_name, node_table_names)
-            self.stdout.write(self.style.SUCCESS(
-                f"{workspace_name} network created"
-            ))
+            Network.create_with_edge_definition(
+                workspace_name, workspace, edge_table_name, node_table_names
+            )
+            self.stdout.write(self.style.SUCCESS(f'{workspace_name} network created'))
 
         except Exception as e:
             self.stderr.write(self.style.ERROR(str(e)))
@@ -110,27 +102,28 @@ def add_arguments(self, parser):
 
     def handle(self, *args, **options):
         # Check that owner exists
-        self.user = User.objects.filter(email=options["email"]).first()
+        self.user = User.objects.filter(email=options['email']).first()
         if self.user is None:
-      	    raise Exception('User did not exist')        
+            raise Exception('User did not exist')
 
         # Create the workspaces
-        self.create_workspace_if_not_exists("boston")
-        self.create_workspace_if_not_exists("eurovis-2019")
-        self.create_workspace_if_not_exists("miserables")
-        self.create_workspace_if_not_exists("movies")
-        self.create_workspace_if_not_exists("openflights")
+        self.create_workspace_if_not_exists('boston')
+        self.create_workspace_if_not_exists('eurovis-2019')
+        self.create_workspace_if_not_exists('miserables')
+        self.create_workspace_if_not_exists('movies')
+        self.create_workspace_if_not_exists('openflights')
 
         # Create the tables
-        self.create_tables_for_workspace("boston", "membership")
-        self.create_tables_for_workspace("eurovis-2019", "connections")
-        self.create_tables_for_workspace("miserables", "relationships")
-        self.create_tables_for_workspace("openflights", "routes")
+        self.create_tables_for_workspace('boston', 'membership')
+        self.create_tables_for_workspace('eurovis-2019', 'connections')
+        self.create_tables_for_workspace('miserables', 'relationships')
+        self.create_tables_for_workspace('openflights', 'routes')
 
         # Create the networks from the tables
-        self.create_network_in_workspace("boston", "membership", ["clubs", "members"])
-        self.create_network_in_workspace("eurovis-2019", "connections", ["people"])
-        self.create_network_in_workspace("miserables", "relationships", ["characters"])
-        self.create_network_in_workspace("openflights", "routes", ["airports"])
-        
-        # TODO: Include movies dataset here once new network creation API is accessible internally from the API
+        self.create_network_in_workspace('boston', 'membership', ['clubs', 'members'])
+        self.create_network_in_workspace('eurovis-2019', 'connections', ['people'])
+        self.create_network_in_workspace('miserables', 'relationships', ['characters'])
+        self.create_network_in_workspace('openflights', 'routes', ['airports'])
+
+        # TODO: Include movies dataset here once new network creation API
+        # is accessible internally from the API