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)