Code Formatting by Docker Black

Author: ayushvjain

backend/api.py

@@ -9,6 +9,7 @@
 from backend.auth import jwt, refresh_token
 from backend.schemas import spec
 from backend.routes.sources import bp as sources_bp
+
 # from backend.routes.incidents import bp as incidents_bp
 from backend.routes.officers import bp as officers_bp
 from backend.routes.agencies import bp as agencies_bp
@@ -48,15 +49,13 @@ def register_extensions(app: Flask):
     # Driver setup
     db_driver = GraphDatabase.driver(
         f'bolt://{app.config["GRAPH_NM_URI"]}',
-        auth=(
-            app.config["GRAPH_USER"],
-            app.config["GRAPH_PASSWORD"]
-        ))
+        auth=(app.config["GRAPH_USER"], app.config["GRAPH_PASSWORD"]),
+    )
 
     try:
         db_driver.verify_connectivity()
-        app.config['DB_DRIVER'] = db_driver
-        neo_config.DRIVER = app.config['DB_DRIVER']
+        app.config["DB_DRIVER"] = db_driver
+        neo_config.DRIVER = app.config["DB_DRIVER"]
         print("Connected to Neo4j")
     except Exception as e:
         print(f"Error connecting to Database: {e}")
@@ -66,15 +65,15 @@ def register_extensions(app: Flask):
     neo_url = "bolt://{user}:{pw}@{uri}".format(
         user=app.config["GRAPH_USER"],
         pw=app.config["GRAPH_PASSWORD"],
-        uri=app.config["GRAPH_NM_URI"]
+        uri=app.config["GRAPH_NM_URI"],
     )
     neo_config.DATABASE_URL = neo_url
 
     spec.register(app)
 
     # Authentication
     ph = PasswordHasher()
-    app.config['PASSWORD_HASHER'] = ph
+    app.config["PASSWORD_HASHER"] = ph
     jwt.init_app(app)
 
     Mail(app)

backend/auth/jwt.py

@@ -24,10 +24,7 @@ def verify_contributor_has_source_or_abort():
     verify_jwt_in_request()
     jwt_decoded = get_jwt()
     current_user = User.get(jwt_decoded["sub"])
-    if (
-        current_user is None
-        or not current_user.member_of
-    ):
+    if current_user is None or not current_user.member_of:
         abort(403)
         return False
     return True

backend/config.py

@@ -1,6 +1,7 @@
 import os
 from dotenv import load_dotenv
 from datetime import timedelta
+
 if os.environ.get("FLASK_ENV") != "production":
     load_dotenv()
 
@@ -34,8 +35,7 @@ class Config(object):
     MAIL_PASSWORD = os.environ.get("MAIL_PASSWORD")
     MAIL_DEFAULT_SENDER = os.environ.get(
         "MAIL_DEFAULT_SENDER",
-        "National Police Data Coalition <{email}>".format(
-              email=MAIL_USERNAME),
+        "National Police Data Coalition <{email}>".format(email=MAIL_USERNAME),
     )
 
     """
@@ -60,18 +60,15 @@ class Config(object):
 
     FRONTEND_PORT = os.environ.get("NPDI_WEB_PORT", "3000")
     FRONTEND_URL = os.environ.get(
-        "FRONTEND_URL",
-        "http://localhost:" + FRONTEND_PORT
+        "FRONTEND_URL", "http://localhost:" + FRONTEND_PORT
     )
 
     SCRAPER_SQS_QUEUE_NAME = os.environ.get("SCRAPER_SQS_QUEUE_NAME")
 
     @property
     def NEO4J_BOLT_URI(self):
         return "bolt://{user}:{pw}@{uri}".format(
-            user=self.GRAPH_USER,
-            pw=self.GRAPH_PASSWORD,
-            uri=self.GRAPH_NM_URI
+            user=self.GRAPH_USER, pw=self.GRAPH_PASSWORD, uri=self.GRAPH_NM_URI
         )
 
     @property

backend/database/core.py

@@ -4,6 +4,7 @@
 Do not import anything directly from `backend.database._core`. Instead, import
 from `backend.database`.
 """
+
 import os
 from typing import Optional
 
@@ -31,11 +32,11 @@ def execute_query(filename: str) -> Optional[List[Dict[str, Any]]]:
     """
     # Read the query from the file
     query_path = os.path.join(QUERIES_DIR, secure_filename(filename))
-    with open(query_path, 'r') as f:
+    with open(query_path, "r") as f:
         query = f.read()
 
     # Get the Neo4j driver
-    neo4j_conn = current_app.config['DB_DRIVER']
+    neo4j_conn = current_app.config["DB_DRIVER"]
 
     # Execute the query using the existing connection
     with neo4j_conn.session() as session:
@@ -76,7 +77,7 @@ def create_database(overwrite: bool):
     """Initialize the Neo4j database by setting up constraints and indexes."""
     if overwrite:
         # Get the Neo4j driver
-        neo4j_conn = current_app.config['DB_DRIVER']
+        neo4j_conn = current_app.config["DB_DRIVER"]
 
         with neo4j_conn.session() as session:
             session.run("MATCH (n) DETACH DELETE n")
@@ -121,14 +122,14 @@ def delete_database(test_db: bool):
     """Delete all data from the Neo4j database."""
     if test_db:
         # If you have a separate test database, drop it
-        test_neo4j_conn = current_app.config['DB_DRIVER']
+        test_neo4j_conn = current_app.config["DB_DRIVER"]
         test_db_name = current_app.config.get("GRAPH_TEST_DB_NAME", "test")
         with test_neo4j_conn.session(database="system") as session:
             session.run(f"DROP DATABASE {test_db_name} IF EXISTS")
         click.echo(f"Test database {test_db_name!r} was deleted.")
     else:
         # Delete all data from the default database
-        neo4j_conn = current_app.config['DB_DRIVER']
+        neo4j_conn = current_app.config["DB_DRIVER"]
         with neo4j_conn.session() as session:
             session.run("MATCH (n) DETACH DELETE n")
         click.echo("Deleted all data from the Neo4j database.")

backend/database/models/agency.py

@@ -9,7 +9,7 @@
     RelationshipTo,
     DateProperty,
     UniqueIdProperty,
-    One
+    One,
 )
 
 
@@ -50,12 +50,17 @@ class Unit(StructuredNode, JsonSerializable):
     agency = RelationshipTo("Agency", "ESTABLISHED_BY", cardinality=One)
     commander = RelationshipTo(
         "backend.database.models.officer.Officer",
-        "COMMANDED_BY", model=UnitMembership)
+        "COMMANDED_BY",
+        model=UnitMembership,
+    )
     officers = RelationshipTo(
         "backend.database.models.officer.Officer",
-        "MEMBER_OF", model=UnitMembership)
+        "MEMBER_OF",
+        model=UnitMembership,
+    )
     citations = RelationshipTo(
-        'backend.database.models.source.Source', "UPDATED_BY", model=Citation)
+        "backend.database.models.source.Source", "UPDATED_BY", model=Citation
+    )
 
     def __repr__(self):
         return f"<Unit {self.name}>"
@@ -79,7 +84,8 @@ class Agency(StructuredNode, JsonSerializable):
     # Relationships
     units = RelationshipTo("Unit", "ESTABLISHED")
     citations = RelationshipTo(
-        'backend.database.models.source.Source', "UPDATED_BY", model=Citation)
+        "backend.database.models.source.Source", "UPDATED_BY", model=Citation
+    )
 
     def __repr__(self):
         return f"<Agency {self.name}>"

backend/database/models/attachment.py

@@ -1,9 +1,5 @@
 from backend.schemas import JsonSerializable
-from neomodel import (
-    StringProperty,
-    UniqueIdProperty,
-    StructuredNode
-)
+from neomodel import StringProperty, UniqueIdProperty, StructuredNode
 
 
 class Attachment(JsonSerializable, StructuredNode):

backend/database/models/civilian.py

@@ -1,9 +1,10 @@
 """Define the Classes for Civilians."""
+
 from neomodel import (
     StructuredNode,
     StringProperty,
     IntegerProperty,
-    RelationshipTo
+    RelationshipTo,
 )
 
 
@@ -14,6 +15,8 @@ class Civilian(StructuredNode):
 
     # Relationships
     complaints = RelationshipTo(
-        "backend.database.models.complaint.Complaint", "COMPLAINED_OF")
+        "backend.database.models.complaint.Complaint", "COMPLAINED_OF"
+    )
     witnessed_complaints = RelationshipTo(
-        "backend.database.models.complaint.Complaint", "WITNESSED")
+        "backend.database.models.complaint.Complaint", "WITNESSED"
+    )

backend/database/models/complaint.py

@@ -1,4 +1,5 @@
 """Define the Classes for Complaints."""
+
 from backend.schemas import JsonSerializable, PropertyEnum
 from neomodel import (
     StructuredNode,
@@ -7,7 +8,7 @@
     RelationshipTo,
     RelationshipFrom,
     DateProperty,
-    UniqueIdProperty
+    UniqueIdProperty,
 )
 
 
@@ -20,10 +21,7 @@ class RecordType(str, PropertyEnum):
 
 # Neo4j Models
 class BaseSourceRel(StructuredRel, JsonSerializable):
-    record_type = StringProperty(
-        choices=RecordType.choices(),
-        required=True
-    )
+    record_type = StringProperty(choices=RecordType.choices(), required=True)
 
 
 class LegalSourceRel(BaseSourceRel):

backend/database/models/litigation.py

@@ -5,7 +5,7 @@
     StringProperty,
     RelationshipTo,
     DateProperty,
-    UniqueIdProperty
+    UniqueIdProperty,
 )
 
 
@@ -35,7 +35,8 @@ class Litigation(StructuredNode, JsonSerializable):
     dispositions = RelationshipTo("Disposition", "YIELDED")
     defendants = RelationshipTo("Officer", "NAMED_IN")
     citations = RelationshipTo(
-        'backend.database.models.source.Source', "UPDATED_BY", model=Citation)
+        "backend.database.models.source.Source", "UPDATED_BY", model=Citation
+    )
 
     def __repr__(self):
         return f"<Litigation {self.uid}:{self.case_title}>"

backend/database/models/officer.py

@@ -4,9 +4,13 @@
 
 from neomodel import (
     StructuredNode,
-    RelationshipTo, RelationshipFrom, Relationship,
-    StringProperty, DateProperty,
-    UniqueIdProperty, One
+    RelationshipTo,
+    RelationshipFrom,
+    Relationship,
+    StringProperty,
+    DateProperty,
+    UniqueIdProperty,
+    One,
 )
 
 
@@ -16,20 +20,26 @@ class StateID(StructuredNode, JsonSerializable):
     law enforcement agencies. For example, in New York, this would be
     the Tax ID Number.
     """
+
     id_name = StringProperty()  # e.g. "Tax ID Number"
     state = StringProperty(choices=State.choices())  # e.g. "NY"
     value = StringProperty()  # e.g. "958938"
-    officer = RelationshipFrom('Officer', "HAS_STATE_ID", cardinality=One)
+    officer = RelationshipFrom("Officer", "HAS_STATE_ID", cardinality=One)
 
     def __repr__(self):
         return f"<StateID: Officer {self.officer_id}, {self.state}>"
 
 
 class Officer(StructuredNode, JsonSerializable):
     __property_order__ = [
-        "uid", "first_name", "middle_name",
-        "last_name", "suffix", "ethnicity",
-        "gender", "date_of_birth"
+        "uid",
+        "first_name",
+        "middle_name",
+        "last_name",
+        "suffix",
+        "ethnicity",
+        "gender",
+        "date_of_birth",
     ]
     __hidden_properties__ = ["citations"]
 
@@ -43,19 +53,23 @@ class Officer(StructuredNode, JsonSerializable):
     date_of_birth = DateProperty()
 
     # Relationships
-    state_ids = RelationshipTo('StateID', "HAS_STATE_ID")
+    state_ids = RelationshipTo("StateID", "HAS_STATE_ID")
     units = Relationship(
-        'backend.database.models.agency.Unit', "MEMBER_OF_UNIT")
+        "backend.database.models.agency.Unit", "MEMBER_OF_UNIT"
+    )
     litigation = Relationship(
-        'backend.database.models.litigation.Litigation', "NAMED_IN")
+        "backend.database.models.litigation.Litigation", "NAMED_IN"
+    )
     allegations = Relationship(
-        'backend.database.models.complaint.Allegation', "ACCUSED_OF")
+        "backend.database.models.complaint.Allegation", "ACCUSED_OF"
+    )
     investigations = Relationship(
-        'backend.database.models.complaint.Investigation', "LEAD_BY")
-    commands = Relationship(
-        'backend.database.models.agency.Unit', "COMMANDS")
+        "backend.database.models.complaint.Investigation", "LEAD_BY"
+    )
+    commands = Relationship("backend.database.models.agency.Unit", "COMMANDS")
     citations = RelationshipTo(
-        'backend.database.models.source.Source', "UPDATED_BY", model=Citation)
+        "backend.database.models.source.Source", "UPDATED_BY", model=Citation
+    )
 
     def __repr__(self):
         return f"<Officer {self.element_id}>"