From 0ea4abca33363f0bbdffa181b60beefc247774fa Mon Sep 17 00:00:00 2001 From: Parker Date: Sun, 25 Feb 2024 02:15:31 -0600 Subject: Creation --- app/db.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 app/db.py (limited to 'app/db.py') diff --git a/app/db.py b/app/db.py new file mode 100644 index 0000000..4d41dcc --- /dev/null +++ b/app/db.py @@ -0,0 +1,31 @@ +import sqlalchemy + +engine = sqlalchemy.create_engine('sqlite:///data.db') + +def init_db(): + with engine.begin() as conn: + conn.execute(sqlalchemy.text( + ''' + CREATE TABLE IF NOT EXISTS accounts ( + account_name, PRIMARY KEY (account_name) + ) + ''' + )) + conn.execute(sqlalchemy.text( + ''' + CREATE TABLE IF NOT EXISTS links ( + owner, link, redirect_link, expire_date, + FOREIGN KEY (owner) REFERENCES accounts(account_name), PRIMARY KEY (link) + ) + ''' + )) + conn.execute(sqlalchemy.text( + ''' + CREATE TABLE IF NOT EXISTS records ( + owner, link, timestamp, ip, location, browser, os, user_agent, isp, + FOREIGN KEY (owner) REFERENCES links(owner), + FOREIGN KEY (link) REFERENCES links(link)) + ''' + )) + + conn.commit() \ No newline at end of file -- cgit v1.2.3-70-g09d2