Skip to content

Commit

Permalink
chore: formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
ElJocho committed Jan 16, 2024
1 parent d909b40 commit 0eb111b
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 10 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from firebase_admin import exceptions

from mapswipe_workers import auth
from mapswipe_workers.definitions import CustomError, logger, ProjectType
from mapswipe_workers.definitions import CustomError, ProjectType, logger


def chunks(data: list, size: int = 250) -> Iterable[list]:
Expand Down Expand Up @@ -96,8 +96,8 @@ def delete_project(project_ids: list) -> bool:
pg_db = auth.postgresDB()
try:
project_type = pg_db.retr_query(
f"Select project_type from projects where project_id = %(project_id)s;",
{"project_id": project_id}
"Select project_type from projects where project_id = %(project_id)s;",
{"project_id": project_id},
)[0][0]
project = ProjectType(project_type).constructor
project.delete_from_postgres(project_id)
Expand All @@ -107,5 +107,4 @@ def delete_project(project_ids: list) -> bool:
f"Tried to delete project which does not exist in postgres: {project_id}"
)


return True
4 changes: 1 addition & 3 deletions mapswipe_workers/mapswipe_workers/project_types/project.py
Original file line number Diff line number Diff line change
Expand Up @@ -605,9 +605,7 @@ def delete_from_postgres(cls, project_id):

del p_con
logger.info(
f"{project_id} - "
f"deleted project, groups and tasks "
f"from postgres"
f"{project_id} - " f"deleted project, groups and tasks " f"from postgres"
)

def calc_required_results(self):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,6 @@
import unittest

from mapswipe_workers import auth
from mapswipe_workers.config import FIREBASE_DB
from mapswipe_workers.definitions import CustomError
from mapswipe_workers.firebase_to_postgres import delete_project
from tests.integration import base, set_up, tear_down

Expand All @@ -22,7 +20,6 @@ def setUp(self):
def tearDown(self):
tear_down.delete_test_data(self.project_id)


def verify_postgres_not_empty(self):
pg_db = auth.postgresDB()
sql_query = f"SELECT * FROM tasks WHERE project_id = '{self.project_id}'"
Expand Down

0 comments on commit 0eb111b

Please # to comment.