diff --git a/pyfpdb/GuiAutoImport.py b/pyfpdb/GuiAutoImport.py index 5081022f..4b594d52 100644 --- a/pyfpdb/GuiAutoImport.py +++ b/pyfpdb/GuiAutoImport.py @@ -57,7 +57,7 @@ class GuiAutoImport (threading.Thread): self.inputFile = os.path.join(self.path, file) stat_info = os.stat(self.inputFile) if not self.import_files.has_key(self.inputFile) or stat_info.st_mtime > self.import_files[self.inputFile]: - fpdb_import.import_file_dict(self, self.settings, callHud = True) + self.importer.import_file_dict(self, self.settings, callHud = True) self.import_files[self.inputFile] = stat_info.st_mtime print "GuiAutoImport.import_dir done" @@ -121,6 +121,7 @@ class GuiAutoImport (threading.Thread): def __init__(self, settings, debug=True): """Constructor for GuiAutoImport""" self.settings=settings + self.importer = fpdb_import.Importer() self.server=settings['db-host'] self.user=settings['db-user'] diff --git a/pyfpdb/GuiBulkImport.py b/pyfpdb/GuiBulkImport.py index 0a6bd10d..cd1277b6 100644 --- a/pyfpdb/GuiBulkImport.py +++ b/pyfpdb/GuiBulkImport.py @@ -32,7 +32,7 @@ class GuiBulkImport (threading.Thread): print "BulkImport is not recursive - please select the final directory in which the history files are" else: self.inputFile=self.path+os.sep+file - fpdb_import.import_file_dict(self, self.settings, False) + self.importer.import_file_dict(self, self.settings, False) print "GuiBulkImport.import_dir done" def load_clicked(self, widget, data=None): @@ -69,7 +69,7 @@ class GuiBulkImport (threading.Thread): if os.path.isdir(self.inputFile): self.import_dir() else: - fpdb_import.import_file_dict(self, self.settings, False) + self.importer.import_file_dict(self, self.settings, False) def get_vbox(self): """returns the vbox of this thread""" @@ -83,6 +83,7 @@ class GuiBulkImport (threading.Thread): def __init__(self, db, settings): self.db=db self.settings=settings + self.importer = fpdb_import.Importer() self.vbox=gtk.VBox(False,1) self.vbox.show() diff --git a/pyfpdb/GuiTableViewer.py b/pyfpdb/GuiTableViewer.py index 15a21149..3582971f 100644 --- a/pyfpdb/GuiTableViewer.py +++ b/pyfpdb/GuiTableViewer.py @@ -255,8 +255,9 @@ class GuiTableViewer (threading.Thread): self.failOnError=False self.minPrint=0 self.handCount=0 + self.importer = fpdb_import.Importer() - self.last_read_hand_id=fpdb_import.import_file_dict(self, self.settings, False) + self.last_read_hand_id=importer.import_file_dict(self, self.settings, False) #end def table_viewer.import_clicked def all_clicked(self, widget, data): diff --git a/pyfpdb/HUD_main.py b/pyfpdb/HUD_main.py old mode 100644 new mode 100755 diff --git a/pyfpdb/RegressionTest.py b/pyfpdb/RegressionTest.py index db1a88b9..8a8facc5 100644 --- a/pyfpdb/RegressionTest.py +++ b/pyfpdb/RegressionTest.py @@ -57,19 +57,19 @@ class TestSequenceFunctions(unittest.TestCase): print self.pgdict.query['list_tables'] self.result = self.pg_db.cursor.execute(self.pgdict.query['list_tables']) - self.failUnless(self.result==13, "Number of tables in database incorrect. Expected 13 got " + str(self.result)) + self.failUnless(self.result==13, "Number of tables in database incorrect. Expected 13 got " + str(self.result)) def testMySQLRecreateTables(self): """Test droping then recreating fpdb table schema""" self.mysql_db.recreate_tables() self.result = self.mysql_db.cursor.execute("SHOW TABLES") - self.failUnless(self.result==13, "Number of tables in database incorrect. Expected 13 got " + str(self.result)) + self.failUnless(self.result==13, "Number of tables in database incorrect. Expected 13 got " + str(self.result)) def testPostgresSQLRecreateTables(self): """Test droping then recreating fpdb table schema""" self.pg_db.recreate_tables() self.result = self.pg_db.cursor.execute(self.pgdict.query['list_tables']) - self.failUnless(self.result==13, "Number of tables in database incorrect. Expected 13 got " + str(self.result)) + self.failUnless(self.result==13, "Number of tables in database incorrect. Expected 13 got " + str(self.result)) if __name__ == '__main__': unittest.main() diff --git a/pyfpdb/fpdb.py b/pyfpdb/fpdb.py index 8c8ac5e9..b4df905c 100755 --- a/pyfpdb/fpdb.py +++ b/pyfpdb/fpdb.py @@ -32,6 +32,7 @@ import GuiBulkImport import GuiTableViewer import GuiAutoImport import GuiGraphViewer +import FpdbSQLQueries class fpdb: def tab_clicked(self, widget, tab_name): diff --git a/pyfpdb/fpdb_import.py b/pyfpdb/fpdb_import.py index 9c52d6bc..6fe8e28a 100755 --- a/pyfpdb/fpdb_import.py +++ b/pyfpdb/fpdb_import.py @@ -38,17 +38,22 @@ import fpdb_simple import fpdb_parse_logic from optparse import OptionParser +class Importer: -def import_file(server, database, user, password, inputFile): - self.server=server - self.database=database - self.user=user - self.password=password - self.inputFile=inputFile - self.settings={'imp-callFpdbHud':False} - import_file_dict(self, settings) + def __init__(self): + """Constructor""" -def import_file_dict(options, settings, callHud=False): + + def import_file(self, server, database, user, password, inputFile): + self.server=server + self.database=database + self.user=user + self.password=password + self.inputFile=inputFile + self.settings={'imp-callFpdbHud':False} + self.import_file_dict(self, settings) + + def import_file_dict(self, options, settings, callHud=False): last_read_hand=0 if (options.inputFile=="stdin"): inputFile=sys.stdin @@ -225,4 +230,4 @@ if __name__ == "__main__": (options, sys.argv) = parser.parse_args() settings={'imp-callFpdbHud':False, 'db-backend':2} - import_file_dict(options, settings, False) +# import_file_dict(options, settings, False)