85 lines
2 KiB
Python
85 lines
2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import os, sys
|
||
|
|
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from psycopg2 import connect
|
||
|
|
from psycopg2 import Error
|
||
|
|
|
||
|
|
from lib import parse_config_from_file
|
||
|
|
|
||
|
|
|
||
|
|
USER_ID = 2
|
||
|
|
N_NEW_ORDER_ROWS = 1000000
|
||
|
|
COMMIT_AFTER = 50
|
||
|
|
AMOUNT_PER_ORDER = 1
|
||
|
|
PRODUCT_NAME = "Wasser"
|
||
|
|
DRINK_ID = 1
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
|
||
|
|
|
||
|
|
print("\nGetting config...")
|
||
|
|
|
||
|
|
config_file = Path(Path(os.path.dirname(__file__)).parent / "config" / "config.sh").absolute()
|
||
|
|
|
||
|
|
config = parse_config_from_file(config_file)
|
||
|
|
|
||
|
|
print(f"Commit will be done after every {COMMIT_AFTER} rows.")
|
||
|
|
|
||
|
|
x = input(f"Do you want to add {N_NEW_ORDER_ROWS} rows to the app_order table? (enter 'yes' to continue) ")
|
||
|
|
try:
|
||
|
|
if str(x) != "yes":
|
||
|
|
exit()
|
||
|
|
except ValueError:
|
||
|
|
exit()
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
print("\nConnecting to database...")
|
||
|
|
|
||
|
|
conn = connect(
|
||
|
|
user = config["PGDB_USER"],
|
||
|
|
password = config["PGDB_PASSWORD"],
|
||
|
|
host = config["PGDB_HOST"],
|
||
|
|
port = config["PGDB_PORT"],
|
||
|
|
database = config["PGDB_DB"]
|
||
|
|
)
|
||
|
|
|
||
|
|
cur = conn.cursor()
|
||
|
|
|
||
|
|
for i in range(N_NEW_ORDER_ROWS):
|
||
|
|
|
||
|
|
cur.execute(f"""
|
||
|
|
insert into app_order (datetime, product_name, price_sum, content_litres, drink_id, user_id, amount)
|
||
|
|
values (
|
||
|
|
current_timestamp,
|
||
|
|
'{PRODUCT_NAME}',
|
||
|
|
10.00,
|
||
|
|
0.5,
|
||
|
|
{DRINK_ID},
|
||
|
|
{USER_ID},
|
||
|
|
{AMOUNT_PER_ORDER}
|
||
|
|
)
|
||
|
|
""")
|
||
|
|
|
||
|
|
if i % COMMIT_AFTER == 0 and not i == 0:
|
||
|
|
conn.commit()
|
||
|
|
print(f"\nAdded {i} rows")
|
||
|
|
|
||
|
|
conn.commit()
|
||
|
|
print(f"\nAdded {N_NEW_ORDER_ROWS} rows")
|
||
|
|
|
||
|
|
print("done with db setup.")
|
||
|
|
|
||
|
|
except (Error, Exception) as err:
|
||
|
|
|
||
|
|
print(f"An error occured while connecting to the database {config['PGDB_DB']} at {config['PGDB_HOST']}:\n{err}", file=sys.stderr)
|
||
|
|
|
||
|
|
finally:
|
||
|
|
|
||
|
|
cur.close()
|
||
|
|
conn.close()
|