creating_test_search_data.py 文件源码

def get_records_for_test_db(src_db_path, num_rec):
    with sqlite3.connect(str(src_db_path)) as src_conn:
        src_conn.row_factory = sqlite3.Row
        query_yield = src_conn.execute('''SELECT * FROM moz_places''')
        # all_records = [DBRecord(record) for record in enumerate(query_yield)]
        all_records = query_yield.fetchall()
    chosen_records = random.choices(all_records, k=num_rec)
    chosen_records = [DBRecord(*record) for record in chosen_records]
    return chosen_records