31 lines
971 B
Python
31 lines
971 B
Python
"""Add rsa, abschnitt columns to lv_positionen + abschnitt to positionen."""
|
|
import sqlite3, os
|
|
|
|
db_path = os.path.normpath(os.path.join(os.path.dirname(__file__), '..', 'data', 'aufmass.db'))
|
|
if not os.path.exists(db_path):
|
|
print("No DB found, skipping.")
|
|
exit(0)
|
|
|
|
conn = sqlite3.connect(db_path)
|
|
c = conn.cursor()
|
|
|
|
c.execute("PRAGMA table_info(lv_positionen)")
|
|
cols = [r[1] for r in c.fetchall()]
|
|
|
|
if 'rsa' not in cols:
|
|
c.execute("ALTER TABLE lv_positionen ADD COLUMN rsa VARCHAR(20)")
|
|
print("Added rsa to lv_positionen")
|
|
if 'abschnitt' not in cols:
|
|
c.execute("ALTER TABLE lv_positionen ADD COLUMN abschnitt VARCHAR(100)")
|
|
print("Added abschnitt to lv_positionen")
|
|
|
|
c.execute("PRAGMA table_info(positionen)")
|
|
pcols = [r[1] for r in c.fetchall()]
|
|
if 'abschnitt' not in pcols:
|
|
c.execute("ALTER TABLE positionen ADD COLUMN abschnitt VARCHAR(100)")
|
|
print("Added abschnitt to positionen")
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
print("Migration done.")
|