Compare commits
No commits in common. "d324ddb3c9bda744caedc637c3941117704fe3ba" and "d3134bc6394bcdb4020c9635cf85d533f0238992" have entirely different histories.
d324ddb3c9
...
d3134bc639
3 changed files with 0 additions and 61 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
@ -1,2 +1 @@
|
||||||
*.db
|
*.db
|
||||||
venv/
|
|
||||||
|
|
|
@ -1 +0,0 @@
|
||||||
tabulate==0.8.10
|
|
59
start.py
59
start.py
|
@ -1,59 +0,0 @@
|
||||||
import argparse
|
|
||||||
import sqlite3
|
|
||||||
from tabulate import tabulate
|
|
||||||
|
|
||||||
def parse_args():
|
|
||||||
"""
|
|
||||||
Parses the arguments from the CLI
|
|
||||||
"""
|
|
||||||
parser = argparse.ArgumentParser(description="Testtool Adress Changes")
|
|
||||||
parser.add_argument(
|
|
||||||
'-c', '--create',
|
|
||||||
help='Creates an example database',
|
|
||||||
action='store_true'
|
|
||||||
)
|
|
||||||
parser.add_argument(
|
|
||||||
'-i', '--insert',
|
|
||||||
help='insert some example data',
|
|
||||||
action='store_true'
|
|
||||||
)
|
|
||||||
parser.add_argument(
|
|
||||||
'-p', '--print',
|
|
||||||
choices=['table', 'string'],
|
|
||||||
default='table',
|
|
||||||
help='prints all the cells from the table',
|
|
||||||
)
|
|
||||||
|
|
||||||
return parser.parse_args()
|
|
||||||
|
|
||||||
|
|
||||||
def execute_sql(args):
|
|
||||||
db = sqlite3.connect("test.db")
|
|
||||||
cur = db.cursor()
|
|
||||||
print(args.print)
|
|
||||||
if args.create:
|
|
||||||
print("Creating")
|
|
||||||
with open ('sql/create_db.sql', 'r') as sql_file:
|
|
||||||
cur.executescript(sql_file.read())
|
|
||||||
elif args.insert:
|
|
||||||
print("insert")
|
|
||||||
with open ('sql/insert_data.sql', 'r') as sql_file:
|
|
||||||
cur.executescript(sql_file.read())
|
|
||||||
elif args.print:
|
|
||||||
cur.execute("SELECT * from customers")
|
|
||||||
if args.print == 'table':
|
|
||||||
print(tabulate(cur.fetchall()))
|
|
||||||
else:
|
|
||||||
print(cur.fetchall())
|
|
||||||
db.commit()
|
|
||||||
db.close()
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def main():
|
|
||||||
args = parse_args()
|
|
||||||
execute_sql(args)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
main()
|
|
Loading…
Add table
Add a link
Reference in a new issue