Started python script

This commit is contained in:
caminsha 2022-07-17 16:27:15 +02:00
parent ae9f4e4497
commit d324ddb3c9
2 changed files with 60 additions and 0 deletions

1
requirements.txt Normal file
View file

@ -0,0 +1 @@
tabulate==0.8.10

59
start.py Normal file
View file

@ -0,0 +1,59 @@
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()