Add option to drop the caching tables from the database and exit

Summary:
This can be used to enforce the regeneration of the caching tables with the latest database schema - for instance, by the package manager when upgrading the PyKolab package.

This patch complements D587 in an attempt to fix T2274.

Reviewers: #pykolab_developers

Maniphest Tasks: T2274

Differential Revision: https://git.kolab.org/D593

1 files changed, 16 insertions, 0 deletions

diff --git a/bin/kolab_smtp_access_policy.py b/bin/kolab_smtp_access_policy.py
index be1681d..b351846 100755
--- a/bin/kolab_smtp_access_policy.py
+++ b/bin/kolab_smtp_access_policy.py
@@ -1219,6 +1219,12 @@ def cache_init():
     else:
         engine = create_engine(cache_uri, echo=False)
 
+    if conf.drop_tables:
+        log.info(_("Dropping caching tables from database"))
+        policy_result_table.drop(engine, checkfirst=True)
+        statistic_table.drop(engine, checkfirst=True)
+        return False
+
     try:
         metadata.create_all(engine)
     except sqlalchemy.exc.OperationalError, e:
@@ -1594,6 +1600,13 @@ if __name__ == "__main__":
             _("Access Policy Options")
         )
 
+    access_policy_group.add_option(  "--drop-tables",
+                            dest    = "drop_tables",
+                            action  = "store_true",
+                            default = False,
+                            help    = _("Drop the caching tables from the " + \
+                                "database and exit."))
+
     access_policy_group.add_option(  "--timeout",
                             dest    = "timeout",
                             action  = "store",
@@ -1626,6 +1639,9 @@ if __name__ == "__main__":
 
     policy_requests = {}
 
+    if conf.drop_tables:
+        sys.exit(0)
+
     # Start the work
     while True:
         policy_request = read_request_input()