def user_repo_test(): user_service = UserService() LOGGER.debug("=======USER REPO TEST=======") username = "******" password = "******" first_name = "Muhammad" last_name = "Aftab" email = "*****@*****.**" LOGGER.debug("adding a user") user, msg = user_service.add_user(username, password, first_name, last_name, email) LOGGER.debug(msg) if user is not None: LOGGER.debug("fetching the newly added user") LOGGER.debug(user_repo.find_by_username(user.username)) LOGGER.debug("modifying user") new_user = {"first_name": "Aftab", "last_name": "Khan", "email": "*****@*****.**", "password": "******"} updated_user = user_repo.update_user(username, new_user) LOGGER.debug("the updated user is") LOGGER.debug(updated_user) LOGGER.debug("deleting user") status = user_repo.delete_user(username) LOGGER.debug(status)
def get_all_users(): # get all users if request.method == 'GET': user = UserService.get_all_user() return make_response(jsonify(user), 200) # Create user if request.method == 'POST': user_data = request.get_json() user_id = UserService.add_user(user_data) return make_response(f"{user_id}", 200)
def fake_other(): s = UserService() s.add_user("admin", "1234", "*****@*****.**", 1) s.add_user("editor", "1234", "*****@*****.**", 2) try: con = mysql_pool.get_connection() con.start_transaction() cursor = con.cursor() cursor.execute("INSERT INTO `t_role` VALUES ('1', '管理员');") cursor.execute("INSERT INTO `t_role` VALUES ('2', '新闻编辑');") cursor.execute("INSERT INTO `t_type` VALUES ('1', '要闻');") cursor.execute("INSERT INTO `t_type` VALUES ('2', '体育');") cursor.execute("INSERT INTO `t_type` VALUES ('3', '科技');") cursor.execute("INSERT INTO `t_type` VALUES ('4', '娱乐');") cursor.execute("INSERT INTO `t_type` VALUES ('5', '历史');") con.commit() except Exception as e: if "con" in dir(): con.rollback() print(e) finally: if "con" in dir(): con.close()
def user(self, ansible_json): us = UserService(ansible_json) try: if eq(ansible_json['state'], 'absent'): result = yield us.del_user() elif eq(ansible_json['state'], 'present'): result = yield us.add_user() elif eq(ansible_json['state'], 'update'): result = yield us.update_password() else: result = self.return_json(2, 'Not Supported') except Exception as err: self.write(self.return_json(-1, err.args)) else: self.write(result)
username = input('\n\tusername: '******'\n\tpassword: '******'\n\ttype in your password again: ') if password != re_password: print('password should be same! return to prev step in 3 sec') time.sleep(3) continue email = input('\n\temail: ') result = __role_service.search_list() for i in range(len(result)): role = result[i] print(Fore.LIGHTBLUE_EX, '\n\t{0}.{1}'.format(i+1, role[1])) print(Style.RESET_ALL) opt=input('\n\twhat is your role: ') role_id = result[int(opt)-1][0] __user_service.add_user(username, password, email, role_id) print('\n\tsuccessfully save return to prev step in 3 sec') time.sleep(3) elif opt =='2': page = 1 while True: os.system('cls') count_page = __user_service.search_count_page() result = __user_service.search_user_list(page) for i in range(len(result)): news = result[i] print(Fore.LIGHTBLUE_EX, '\n\t{0}\t{1}\t{2}'.format(i + 1, news[1], news[2])) print(Fore.LIGHTBLUE_EX, '\n\t==================') print(Fore.LIGHTBLUE_EX, '\n\t{0}/{1}'.format(page, count_page))
def device_repo_test(): user_service = UserService() print "\n\n\n=======DEVICE REPO TEST=======" print "\ncreating a test user" user, msg = user_service.add_user("muhaftab", "1234", "Muhammad", "Aftab", "*****@*****.**") print msg print "adding a new device for user: %s" % user.username device1 = device_repo.add_device(user.username, "TableLamp") print "returned device is ", device1 device2 = device_repo.add_device(user.username, "Kettle") print "returned device is ", device2 print "\nfetching the new device from db" device = device_repo.find_device(user.username, device1.device_id)[0] print device print "\nfetching the user to see if device is added for user" user = user_service.get_user(user.username) print [d.serialize() for d in user.devices] print "\nadding some consumption data for the device" c1 = DeviceConsumption(10.0, 0.12, False, datetime.datetime.now()) c2 = DeviceConsumption(11.0, 2.12, True, datetime.datetime.now()) device_repo.add_device_consumption(device, c1) device_repo.add_device_consumption(device, c2) print "\ntesting if consumption data is added to device" print [c.serialize() for c in device.consumption] print "\nModifying device" new_device = Device("CoffeeMaker") updated_device = device_repo.update_device(user.username, device.device_id, new_device) print updated_device print "\ntesting if consumption data exists for updated device" print [c.serialize() for c in updated_device.consumption] print "\nadding device model to the device" json_params = {"p_peak": 80.8, "p_stable": 50.0, "lambda": 0.31} m1 = DeviceModel("ExponentialDecay", json_params) device_repo.add_device_model(updated_device, m1) print updated_device.serialize() print "\nfetching the user again to see if updated device is shown" user = user_service.get_user(user.username) print user.serialize() print "\nfetching list of devices for the user" print [device.serialize() for device in user_service.get_devices(user.username)] print "\ndeleting device" status = device_repo.delete_device(user.username, device.device_id) print status print "\nfetching the user agian to see if device is indeed deleted" user = user_service.get_user(user.username) print [d for d in user.devices] print "\nfinally deleting user" status = user_service.delete_user(user.username) print status