def process_uploaded_file(f, name):
partial_path = "mysite/uploads/%s" % name
uploaded_file = os.path.join(settings.BASE_DIR, partial_path)
with open(uploaded_file, 'wb+') as destination:
for chunk in f.chunks():
destination.write(chunk)
ft = FileTable(uploaded_file)
ft.CreateAndPopulateTable()
def testRemoveUploadedFile(self):
return
dba = DBAccess(db_path=th.TEST_DB_PATH)
ft = FileTable(th.FILE_PATH, db=th.TEST_DB_PATH)
ft.CreateAndPopulateTable()
table_set = set(FileTable.DataTableList(db=th.TEST_DB_PATH))
self.assertTrue(table_set.issuperset(set([ft._table_name])))
FileTable.RemoveUploadedFile(ft._filename, db=th.TEST_DB_PATH)
remove_table_set = set(FileTable.DataTableList(db=th.TEST_DB_PATH))
self.assertFalse(remove_table_set.issuperset(set([ft._table_name])))
def testCreateAndPopulateTable(self):
return
ft = FileTable(th.FILE_PATH, db=th.TEST_DB_PATH)
ft.CreateAndPopulateTable()
conn = sqlite3.connect(th.TEST_DB_PATH)
sql_str = "SELECT * FROM %s" % ft._table_name
rows = th.ExecuteSQL(conn, sql_str)
self.assertEqual(len(rows), 3)
row = rows[1]
self.assertEqual(row[0], 2)
def testDataTableList(self):
return
dba = DBAccess(db_path=th.TEST_DB_PATH)
ft = FileTable(th.FILE_PATH, db=th.TEST_DB_PATH)
sql_str = "DELETE FROM %s WHERE table_name='%s'" % (
settings.UPLOAD_FILE_TABLE, ft._table_name)
try:
dba.ExecuteQuery(sql_str, commit=True)
except:
pass
old_table_set = set(FileTable.DataTableList(db=th.TEST_DB_PATH))
ft.CreateAndPopulateTable()
new_table_set = set(FileTable.DataTableList(db=th.TEST_DB_PATH))
new_row_set = set([ft._table_name])
self.assertTrue(new_table_set.issuperset(new_row_set))
self.assertFalse(old_table_set.issuperset(new_row_set))
ft._ExecuteSQL('', CloseDB=True)
