def find_file(file_item): """ Finds the file in the database :param file_item: :return: the file_item present in the database """ conn = create_connection() with conn: cursor = conn.execute(__SQL_SELECT__, (file_item.file_name, file_item.file_path)) return [FileItem.from_db(*row) for row in cursor.fetchall()]
def testCreateFromDB(self): now = datetime.now() item = FileItem.from_db("NAME", "PATH", 123, FileItem.FILE_STATUS_INIT, now, now, "ENC-NAME", "ENC-PATH", 12, "CODEC") self.assertEqual(item.file_name, "NAME") self.assertEqual(item.file_path, "PATH") self.assertEqual(item.file_size, 123) self.assertEqual(item.file_status, FileItem.FILE_STATUS_INIT) self.assertEqual(item.start_time, now) self.assertEqual(item.end_time, now) self.assertEqual(item.encoded_file_name, "ENC-NAME") self.assertEqual(item.encoded_file_path, "ENC-PATH") self.assertEqual(item.encoded_file_size, 12) self.assertEqual(item.codec, "CODEC")