Make table names global variables

next-exp · Mar 10, 2022 · 58a97c6 · 58a97c6
1 parent 330460f
commit 58a97c6
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/invisible_cities/database/download.py b/invisible_cities/database/download.py
@@ -71,18 +71,17 @@ def loadDB(dbname : str, tables : list):
         copy_all_rows(conn_sqlite, cursor_sqlite, cursor_mysql, table)
 
 
-if __name__ == '__main__':
-
-    dbnames        = ('NEWDB', 'DEMOPPDB', 'NEXT100DB', 'Flex100DB')
-    common_tables  = ('DetectorGeo','PmtBlr','ChannelGain','ChannelMapping','ChannelMask',
-                      'PmtNoiseRms','ChannelPosition','SipmBaseline', 'SipmNoisePDF',
-                      'PMTFEMapping', 'PMTFELowFrequencyNoise')
-    extended       = dict(NEXT100DB = ("Activity", "Efficiency")) 
+dbnames        = ('NEWDB', 'DEMOPPDB', 'NEXT100DB', 'Flex100DB')
+common_tables  = ('DetectorGeo','PmtBlr','ChannelGain','ChannelMapping','ChannelMask',
+                  'PmtNoiseRms','ChannelPosition','SipmBaseline', 'SipmNoisePDF',
+                  'PMTFEMapping', 'PMTFELowFrequencyNoise')
+extended       = dict(NEXT100DB = ("Activity", "Efficiency"))
 
-    table_dict = dict.fromkeys(dbnames, commmon_tables)
-    for dbname, extra in extended.items():
-        table_dict[dbname] += extra
+table_dict = dict.fromkeys(dbnames, common_tables)
+for dbname, extra in extended.items():
+    table_dict[dbname] += extra
 
+if __name__ == '__main__':
     if len(sys.argv) > 1:
         dbname = sys.argv[1]
         loadDB(dbname, table_dict[dbname])