darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-08-20 14:29:51 -04:00
Code Issues Packages Projects Releases Wiki Activity

broadcast-areas: refactor to use repo.py

Browse Source 
share functionality between repo used by create script and by package

Signed-off-by: Toby Lorne <toby.lornewelch-richards@digital.cabinet-office.gov.uk>
This commit is contained in:
Toby Lorne
2020-07-24 16:45:41 +01:00
parent 15d4d3c0a1
commit 9a9bf3c270
3 changed files with 161 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
notifications_utils/broadcast_areas/__init__.py
View File

@@ -9,59 +9,7 @@ from notifications_utils.formatters import formatted_list
from notifications_utils.serialised_model import SerialisedModelCollection
from notifications_utils.safe_string import make_string_safe_for_id
class BroadcastAreasRepository(object):
def __init__(self):
self.database = Path(__file__).resolve().parent / 'broadcast-areas.sql'
def query(self, sql, *args):
package_path = Path(__file__).resolve().parent
db_filepath = package_path / "broadcast-areas.sqlite3"
with sqlite3.connect(str(db_filepath)) as conn:
cursor = conn.cursor()
cursor.execute(sql, (*args,))
return cursor.fetchall()
def get_datasets(self):
q = "SELECT DISTINCT(dataset) AS dataset FROM broadcast_areas"
datasets = [row[0] for row in self.query(q)]
return sorted(datasets)
def get_dataset_description(self, dataset):
q = """
WITH
areas AS (SELECT * FROM broadcast_areas WHERE dataset = ?),
area_count AS (SELECT COUNT(*) AS c FROM areas),
subset_area_count AS (SELECT c - 4 FROM area_count),
some_area_names AS (SELECT name FROM areas LIMIT 100),
some_shuffled_area_names AS (
SELECT name FROM some_area_names ORDER BY RANDOM()
),
description_area_names AS (
SELECT name FROM some_shuffled_area_names LIMIT 4
),
description_areas_joined AS (
SELECT GROUP_CONCAT(name, ", ") FROM description_area_names
)
SELECT
CASE (SELECT * FROM subset_area_count)
WHEN 0 THEN
(SELECT * FROM description_areas_joined)
ELSE
(SELECT * FROM description_areas_joined)
|| ", "
|| (SELECT * FROM subset_area_count)
|| " more…"
END
"""
description = self.query(q, dataset)[0][0]
return description
def get_areas(self):
pass
def get_areas_from_list(self, areas):
pass
from .repo import BroadcastAreasRepository
@lru_cache(maxsize=128)