10
1
mirror of https://github.com/pfloos/quack synced 2024-09-27 20:11:05 +02:00
QuAcK/tests/create_database.py
Abdallah Ammar 9aebbe74f0 saving
2024-08-29 20:48:16 +02:00

26 lines
554 B
Python

import sqlite3
from molecule import save_molecules_to_json, load_molecules_from_json
from molecule import create_database, add_molecule_to_db
from feather_bench import FeatherBench
# Save molecules to JSON
save_molecules_to_json(FeatherBench, 'FeatherBench.json')
# Load molecules from JSON
loaded_molecules = load_molecules_from_json('FeatherBench.json')
print(loaded_molecules)
# Create a database and add molecules
db_name = 'FeatherBench.db'
create_database(db_name)
for molecule in FeatherBench:
add_molecule_to_db(db_name, molecule)