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')
