def test_get_groups(self): group1 = Group.create_group(name="group1") group2 = Group.create_group(name="group2") user1 = User.create_user(username="******", email="*****@*****.**", password="******") Member.add_member_by_email(group1.G_Id, "*****@*****.**") Member.add_member_by_email(group2.G_Id, "*****@*****.**") test = Member.get_groups(user1[0].U_Id) self.assertEqual(test.count(), 2) Member.delete_member(group1.G_Id, user1[0].U_Id) test = Member.get_groups(user1[0].U_Id) self.assertEqual(test.count(), 1)
def store_add_post(): """ store 添加商店 """ db = g._imdb developer_id = session['user']['id'] appid = config.KEFU_APPID name = request.form['name'] if request.form.has_key('name') else None if not name: return INVALID_PARAM() db.begin() gid = Group.create_group(db, appid, 0, name, False) store_id = Store.create_store(db, name, gid, developer_id) db.commit() #将名称存储redis,用于后台推送 Store.set_store_name(g.im_rds, store_id, name) content = "%d,%d,%d" % (gid, appid, 0) publish_message(g.im_rds, "group_create", content) obj = {"store_id": store_id} return redirect(url_for('.store_index'))
def add_store(): db = g._imdb developer_id = g.developer_id appid = KEFU_APPID name = request.form['name'] if request.form.has_key('name') else None if not name: return INVALID_PARAM() mode = default_mode db.begin() if mode == MODE_BROADCAST: gid = Group.create_group(db, appid, 0, name, False) else: gid = 0 store_id = Store.create_store(db, name, gid, mode, developer_id) db.commit() #将名称存储redis,用于后台推送 Store.set_store_name(g.im_rds, store_id, name) if gid: content = "%d,%d,%d"%(gid, appid, 0) publish_message(g.im_rds, "group_create", content) obj = {"store_id":store_id} return make_response(200, obj)
def store_add_post(): """ store 添加商店 """ db = g._imdb developer_id = session['user']['id'] appid = config.KEFU_APPID name = request.form['name'] if request.form.has_key('name') else None if not name: return INVALID_PARAM() mode = default_mode db.begin() if mode == MODE_BROADCAST: gid = Group.create_group(db, appid, 0, name, False) else: gid = 0 store_id = Store.create_store(db, name, gid, mode, developer_id) db.commit() #将名称存储redis,用于后台推送 Store.set_store_name(g.im_rds, store_id, name) if gid: content = "%d,%d,%d"%(gid, appid, 0) publish_message(g.im_rds, "group_create", content) obj = {"store_id":store_id} return redirect(url_for('.store_index'))
def create_or_read_group(group_id=None): from models import Group if group_id == None and request.method == 'GET': return Group.get_groups() elif group_id == None: group_name = request.json['group_name'] return Group.create_group(group_name) else: return Group.get_group(group_id)
def managegroup(): form = GroupForm() group_show = Group() info = group_show.show_group_own(current_user.user_email) group_id = request.form.get('group_id') print(group_id) print(form.validate_on_submit()) if form.validate_on_submit(): print("1111\n") group_id = request.form.get('group_id') user = current_user.user_email print(group_id, user) group_new = Group() group_new.create_group(user, group_id) return redirect(request.args.get('next') or url_for('managegroup')) return render_template('managegroup.html', groups=info, form=form, user_info_global=user_info_global)
def add_group(): group_info = request.json new_group = Group.create_group(group_info["group_name"]) db.session.add(new_group) try: db.session.commit() # serialize contact and return 201 return f"{group_info['group_name']} group added succesfully", 201 except Exception as error: db.session.rollback() print(f"{error.args} {type(error)}") # return error return jsonify({"result": f"{error.args}"}), 500
def create_group(id): user = check_user_session(id) if not user: return redirect('/') group_form = GroupForm() if group_form.validate_on_submit(): name = group_form.name.data group = Group(name=name, user_id=user.id) group = Group.create_group(group) flash(f'Group created: {name}') return redirect(f'/users/{user.id}') else: return render_template('/users/group.html', form=group_form, user=user)
def test_delete_transaction_by_date(self): group1 = Group.create_group(name="group1") user1 = User.create_user(username="******", email="*****@*****.**", password="******") Member.add_member_by_email(group1.G_Id, "*****@*****.**") Group_Transaction.add_transaction(group1.G_Id, user1[0].U_Id, "Food", "Panda Express", 10.00, '2017-04-04 08:00:01') Group_Transaction.add_transaction(group1.G_Id, user1[0].U_Id, "Food", "Pizza Hut", -10.00, self.current_time) test = Group_Transaction.get_user_transactions(group1.G_Id, user1[0].U_Id) self.assertEqual(len(test), 2) Group_Transaction.delete_transaction_by_date('2017-04-04 08:00:01') test = Group_Transaction.get_user_transactions(group1.G_Id, user1[0].U_Id) self.assertEqual(len(test), 1)
def test_get_members(self): group1 = Group.create_group(name="group1") user1 = User.create_user(username="******", email="*****@*****.**", password="******") User.create_user(username="******", email="*****@*****.**", password="******") User.create_user(username="******", email="*****@*****.**", password="******") Member.add_member_by_email(group1.G_Id, "*****@*****.**") Member.add_member_by_email(group1.G_Id, "*****@*****.**") Member.add_member_by_email(group1.G_Id, "*****@*****.**") test = Member.get_members(group1.G_Id) self.assertEqual(len(test), 3) test = Member.delete_member(group1.G_Id, user1[0].U_Id) self.assertEqual(len(test), 2)
def test_get_user_transactions(self): group1 = Group.create_group(name="group1") user1 = User.create_user(username="******", email="*****@*****.**", password="******") Member.add_member_by_email(group1.G_Id, "*****@*****.**") transaction1 = Group_Transaction.add_transaction(group1.G_Id, user1[0].U_Id, "Food", "Panda Express", 10.00, self.current_time) Group_Transaction.add_transaction(group1.G_Id, user1[0].U_Id, "Food", "Pizza Hut", -10.00, self.current_time) Group_Transaction.add_transaction(group1.G_Id, user1[0].U_Id, "Food", "Arbys", 5.00, self.current_time) self.assertEqual(len(Group_Transaction.get_user_transactions(group1.G_Id, user1[0].U_Id)), 3) Group_Transaction.delete_transaction(group1.G_Id, transaction1.GT_Id) self.assertEqual(len(Group_Transaction.get_user_transactions(group1.G_Id, user1[0].U_Id)), 2)
def test_delete_group(self): group1 = Group.create_group(name="group1") user1 = User.create_user(username="******", email="*****@*****.**", password="******") User.create_user(username="******", email="*****@*****.**", password="******") User.create_user(username="******", email="*****@*****.**", password="******") Member.add_member_by_email(group1.G_Id, "*****@*****.**") Member.add_member_by_email(group1.G_Id, "*****@*****.**") Member.add_member_by_email(group1.G_Id, "*****@*****.**") Group_Transaction.add_transaction(1, 1, "Food", "Panda Express", 20.00, self.current_time) Group_Transaction.add_transaction(1, 2, "Food", "Panda Express", -10.00, self.current_time) Group_Transaction.add_transaction(1, 3, "Food", "Panda Express", -10.00, self.current_time) Group.delete_group(group1.G_Id, group1.G_Name) self.assertEqual(len(Group_Transaction.get_user_transactions(group1.G_Id, user1[0].U_Id)), 0)
def expocr_group_create(request): if request.method == 'GET': params = request.GET elif request.method == 'POST': params = request.POST name = params.get('group_name') print ("G Name from views.py") print (name) result = Group.create_group(name) data = {} gid = result.G_Id data['g_id'] = int(result.G_Id) data['g_name'] = str(result.G_Name) uid = params.get('u_id') emails = params.get('group_members_emails').split(',') uid_res = Member.add_member_by_id(gid, uid) # emails = data['emails'].split(',') for email in emails: email = email.strip() if email is not "": Member.add_member_by_email(gid,email) response = HttpResponse(json.dumps(data), content_type='application/json') return response
def test_get_group_transactions(self): group1 = Group.create_group(name="group1") user1 = User.create_user(username="******", email="*****@*****.**", password="******") user2 = User.create_user(username="******", email="*****@*****.**", password="******") user3 = User.create_user(username="******", email="*****@*****.**", password="******") Member.add_member_by_email(group1.G_Id, "*****@*****.**") Member.add_member_by_email(group1.G_Id, "*****@*****.**") Member.add_member_by_email(group1.G_Id, "*****@*****.**") test = Group_Transaction.get_group_transactions(group1.G_Id) self.assertEqual(test.count(), 0) Group_Transaction.add_transaction(group1.G_Id, user1[0].U_Id, "Food", "Panda Express", 20.00, self.current_time) Group_Transaction.add_transaction(group1.G_Id, user2[0].U_Id, "Food", "Panda Express", -10.00, self.current_time) Group_Transaction.add_transaction(group1.G_Id, user3[0].U_Id, "Food", "Panda Express", -10.00, self.current_time) test2 = Group_Transaction.get_group_transactions(group1.G_Id) self.assertEqual(test2.count(), 3)
def test_update_group_name(self): test = Group.create_group('group3') self.assertEqual(test.G_Name, 'group3') Group.update_group_name(g_id=test.G_Id, g_name='group111') test = Group.get_group_name(id=test.G_Id) self.assertEqual(test[0].G_Name, 'group111')