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)