ElectionDay: Adds support for eCH-0252 V2 and views for municipality results

TYPE: Feature
LINK: ogc-3103

Author: Tschuppi81

.gitignore

@@ -108,3 +108,5 @@ dump.rdb
 .codebuddy/
 
 .aider*
+
+.DS_Store

setup.cfg

@@ -174,7 +174,7 @@ install_requires =
     xlrd
     xlsxwriter!=3.2.4
     xsdata!=23.5
-    xsdata-ech<0.4.0
+    xsdata-ech
     yubico-client
     zope.sqlalchemy
 

src/onegov/election_day/cli.py

@@ -9,6 +9,9 @@
 from onegov.core.cli import pass_group_context
 from onegov.core.sms_processor import SmsQueueProcessor
 from onegov.election_day.collections import ArchivedResultCollection
+from onegov.election_day.collections import ElectionCollection
+from onegov.election_day.collections import ElectionCompoundCollection
+from onegov.election_day.collections import VoteCollection
 from onegov.election_day.models import ArchivedResult
 from onegov.election_day.models import Subscriber
 from onegov.election_day.utils import add_local_results
@@ -250,3 +253,41 @@ def migrate(request: ElectionDayRequest, app: ElectionDayApp) -> None:
         click.echo(f'Migrated {count} subscribers')
 
     return migrate
+
+
+@cli.command('delete-elections-and-votes')
+@click.option('--confirm', is_flag=True, default=False,
+              help='Required to actually delete.')
+def delete_elections_and_votes(confirm: bool) -> Processor:
+    r""" Deletes all elections, election compounds, votes and archived results.
+
+    .. code-block:: bash
+
+        onegov-election-day --select '/onegov_election_day/sg' \
+            delete-elections-and-votes --confirm
+
+    """
+
+    def delete(request: ElectionDayRequest, app: ElectionDayApp) -> None:
+        if not confirm:
+            click.secho(
+                'Dry run — pass --confirm to actually delete.', fg='yellow'
+            )
+            return
+
+        session = request.app.session()
+        click.secho(f'Deleting elections/votes for {app.schema}', fg='yellow')
+
+        for election in ElectionCollection(session).query():
+            session.delete(election)
+        for compound in ElectionCompoundCollection(session).query():
+            session.delete(compound)
+        for vote in VoteCollection(session).query():
+            session.delete(vote)
+        session.query(ArchivedResult).filter_by(
+            schema=session.info['schema']
+        ).delete()
+
+        click.secho('Done.', fg='green')
+
+    return delete

src/onegov/election_day/collections/__init__.py

@@ -2,6 +2,14 @@
 
 from onegov.election_day.collections.archived_results import \
     ArchivedResultCollection
+from onegov.election_day.collections.archived_results import \
+    AllMunicipalArchivedResultCollection
+from onegov.election_day.collections.archived_results import \
+    MunicipalArchivedResultCollection
+from onegov.election_day.collections.archived_results import \
+    MunicipalityArchivedResultCollection
+from onegov.election_day.collections.archived_results import \
+    MunicipalityYearArchivedResultCollection
 from onegov.election_day.collections.archived_results import \
     SearchableArchivedResultCollection
 from onegov.election_day.collections.ballots import BallotCollection
@@ -34,6 +42,10 @@
     'ElectionCompoundCollection',
     'EmailSubscriberCollection',
     'ListCollection',
+    'AllMunicipalArchivedResultCollection',
+    'MunicipalArchivedResultCollection',
+    'MunicipalityArchivedResultCollection',
+    'MunicipalityYearArchivedResultCollection',
     'NotificationCollection',
     'ScreenCollection',
     'SearchableArchivedResultCollection',