importer revisited - sfa-import.py sfa-nuke.py (no more -plc)

[sfa.git] / sfa / importer / sfa-nuke.py

diff --git a/sfa/importer/sfa-nuke.py b/sfa/importer/sfa-nuke.py
new file mode 100755 (executable)
index 0000000..a4967c3
--- /dev/null
+++ b/sfa/importer/sfa-nuke.py
@@ -0,0 +1,69 @@
+#!/usr/bin/python
+##
+# Delete all the database records for SFA. This tool is used to clean out SFA
+# records during testing.
+#
+# Authority info (maintained by the hierarchy module in a subdirectory tree)
+# is not purged by this tool and may be deleted by a command like 'rm'.
+##
+
+import sys
+import os
+from optparse import OptionParser
+
+from sfa.util.sfalogging import logger
+
+from sfa.storage.alchemy import engine
+from sfa.storage.dbschema import DBSchema
+
+def main():
+   usage="%prog: trash the registry DB"
+   parser = OptionParser(usage=usage)
+   parser.add_option("-f","--file-system",dest='clean_fs',action='store_true',default=False,
+                     help="Clean up the /var/lib/sfa/authorities area as well")
+   parser.add_option("-c","--certs",dest='clean_certs',action='store_true',default=False,
+                     help="Remove all cached certs/gids found in /var/lib/sfa/authorities area as well")
+   parser.add_option("-0","--no-reinit",dest='reinit',action='store_false',default=True,
+                     help="By default a new DB schema is installed after the cleanup; this option prevents that")
+   (options,args)=parser.parse_args()
+   if args:
+      parser.print_help()
+      sys.exit(1)
+   dbschema=DBSchema()
+   logger.info("Purging SFA records from database")
+   dbschema.nuke()
+   # for convenience we re-create the schema here, so there's no need for an explicit
+   # service sfa restart
+   # however in some (upgrade) scenarios this might be wrong
+   if options.reinit:
+      logger.info("re-creating empty schema")
+      dbschema.init_or_upgrade()
+
+   if options.clean_certs:
+      # remove the server certificate and all gids found in /var/lib/sfa/authorities
+      logger.info("Purging cached certificates")
+      for (dir, _, files) in os.walk('/var/lib/sfa/authorities'):
+         for file in files:
+            if file.endswith('.gid') or file == 'server.cert':
+               path=dir+os.sep+file
+               os.unlink(path)
+               if not os.path.exists(path):
+                  logger.info("Unlinked file %s"%path)
+               else:
+                  logger.error("Could not unlink file %s"%path)
+
+   if options.clean_fs:
+      # just remove all files that do not match 'server.key' or 'server.cert'
+      logger.info("Purging registry filesystem cache")
+      preserved_files = [ 'server.key', 'server.cert']
+      for (dir,_,files) in os.walk('/var/lib/sfa/authorities'):
+         for file in files:
+            if file in preserved_files: continue
+            path=dir+os.sep+file
+            os.unlink(path)
+            if not os.path.exists(path):
+               logger.info("Unlinked file %s"%path)
+            else:
+               logger.error("Could not unlink file %s"%path)
+if __name__ == "__main__":
+   main()