def login(*, username):
if not in_database(table='BaseUser', user_name=username):
print('Unknown user!')
new_user = input('Would you like to create new user? ')
if new_user in ['yes', 'y']:
type_user = input('Are you a Client or Mechanic? ')
name = input('Provide user_name: ')
phone_number = input('Provide phone_number: ')
email = input('Provide email: ')
address = input('Provide address: ')
if type_user.lower() == 'client':
add_new_client(user_name=name,
phone_number=phone_number,
email=email,
address=address)
output_for_new_client(user_name=name)
menu_for_client(user_name=username)
elif type_user.lower() == 'mechanic':
title = input('Enter your title: ')
add_new_mechanic(user_name=name,
phone_number=phone_number,
email=email,
address=address,
title=title)
output_for_new_client(user_name=name)
# menu(user_name=user_name)
else:
customer_id = user_id(user_name=username)
if in_database(table='Client', base_id=customer_id):
menu_for_client(user_name=username)
elif in_database(table='Mechanic', base_id=customer_id):
menu_for_mechanic(user_name=username)
def add_new_mechanic(*, user_name, phone_number, email, address, title):
connection = sqlite3.connect('vehicle_repair_manager.db')
cursor = connection.cursor()
add_into_base_user = '''
INSERT INTO BaseUser(user_name, phone_number, email, address) VALUES(?, ?, ?, ?)
'''
cursor.execute(add_into_base_user,
(user_name, phone_number, email, address))
connection.commit()
user = user_id(user_name=user_name)
add_into_mechanic = '''
INSERT INTO Mechanic(base_id, title) VALUES(?, ?)
'''
cursor.execute(add_into_mechanic, (user, title))
connection.commit()
connection.close()
def add_new_client(*, user_name, phone_number, email, address):
connection = sqlite3.connect('vehicle_repair_manager.db')
cursor = connection.cursor()
add_into_base_user = '''
INSERT INTO BaseUser(user_name, phone_number, email, address) VALUES(?, ?, ?, ?)
'''
cursor.execute(add_into_base_user,
(user_name, phone_number, email, address))
connection.commit()
user = user_id(user_name=user_name)
add_into_client = '''
INSERT INTO Client(base_id) VALUES(?)
'''
cursor.execute(add_into_client, (user, ))
connection.commit()
connection.close()
def list_personal_vehicles(*, user_name):
connection = sqlite3.connect('vehicle_repair_manager.db')
cursor = connection.cursor()
all_vehicles = '''
SELECT *
FROM Vehicle
WHERE owner = (?)
'''
owner = user_id(user_name=user_name)
cursor.execute(all_vehicles, (owner, ))
connection.commit()
vehicles = cursor.fetchall()
print(
tabulate([x for x in vehicles],
headers=[
'id', 'Category', 'Make', 'Model', 'Register Number',
'Gear Box', 'OwnerId'
]))
def add_vehicle(*, user_name):
category = input('Vehicle category: ')
make = input('Vehicle make: ')
model = input('Vehicle model: ')
register_number = input('Vehicle register number: ')
gear_box = input('Vehicle gear box: ')
connection = sqlite3.connect('vehicle_repair_manager.db')
cursor = connection.cursor()
new_vehicle = '''
INSERT INTO Vehicle(category, make, model, register_number, gear_box, owner)
VALUES(?, ?, ?, ?, ?, ?)
'''
client = user_id(user_name=user_name)
cursor.execute(new_vehicle,
(category, make, model, register_number, gear_box, client))
connection.commit()
connection.close()
print('Thank you! You added new personal vehicle!')
def update_vehicle(*, vehicle_id):
connection = sqlite3.connect('vehicle_repair_manager.db')
cursor = connection.cursor()
print('What do you want to update? ')
new_category = input('New category: ')
new_make = input('New made: ')
new_model = input('New model: ')
new_reg_num = input('New register number: ')
new_gear_box = input('New gear box: ')
new_owner = input('New owner id: ')
cursor.execute('''SELECT * FROM Vehicle WHERE id = (?)''', (vehicle_id, ))
new_data = []
vehicle_info = cursor.fetchone()
if new_category != '':
new_data.append(new_category)
else:
new_category = vehicle_info[1]
new_data.append(new_category)
if new_make != '':
new_data.append(new_make)
else:
new_make = vehicle_info[2]
new_data.append(new_make)
if new_model != '':
new_data.append(new_model)
else:
new_model = vehicle_info[3]
new_data.append(new_model)
if new_reg_num != '':
new_data.append(new_reg_num)
else:
new_reg_num = vehicle_info[4]
new_data.append(new_reg_num)
if new_gear_box != '':
new_data.append(new_gear_box)
else:
new_gear_box = vehicle_info[5]
new_data.append(new_gear_box)
if new_owner != '':
new_owner_id = user_id(user_name=new_owner)
new_data.append(new_owner_id)
else:
new_owner = vehicle_info[6]
new_data.append(new_owner)
new_data.append(vehicle_id)
update_data = '''
UPDATE Vehicle
SET category = (?),
make = (?),
model = (?),
register_number = (?),
gear_box = (?),
owner = (?)
WHERE id = (?)
'''
cursor.execute(update_data, tuple(new_data))
connection.commit()
connection.close()
print('You update the vehicle successfully!')
def select():
if request.method == "GET":
uid = request.args.get('id')
res = utils.user_id(table, field, uid)
return render_template('update.html', res=res)