10
1
mirror of https://github.com/pfloos/quack synced 2024-09-27 20:11:05 +02:00
QuAcK/tests/create_database.py
2024-08-29 09:45:30 +02:00

26 lines
526 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 swift_set import swiftset
# Save molecules to JSON
save_molecules_to_json(swiftset, 'swiftset.json')
# Load molecules from JSON
loaded_molecules = load_molecules_from_json('swiftset.json')
print(loaded_molecules)
# Create a database and add molecules
db_name = 'swiftset.db'
create_database(db_name)
for molecule in swiftset:
add_molecule_to_db(db_name, molecule)