def save(self,id=None): new_category = Category(name=self.cleaned_data['name'], p_category=self.cleaned_data['p_category'], user = self.user, isIncome=self.cleaned_data['isIncome'], id = id) new_category.save()
def init_data(user): """this is init""" category1 = Category(p_category=None, name="cc", isIncome=True, user=user) category1.save() item1 = Item.objects.create(price=100.00, comment="", category=category1, pub_date="2015-05-24") return item1, category1
def test_data(): pwd = '123456' name1 = 'xiaoming' name2 = 'xiaohong' u1 = User.objects.filter(username=name1) u2 = User.objects.filter(username=name2) u1 = u1 if u1.exists() else User.objects.create_user(username=name1, password=pwd) u2 = u2 if u2.exists() else User.objects.create_user(username=name2, password=pwd) u_list = User.objects.filter(username__contains='xiao') print(u_list) for u in u_list: c1 = Category(name='salary', isIncome=True, p_category=None, user=u) c1.save() c2 = Category(name='salary_lp', isIncome=True, p_category=c1, user=u) c2.save() base_salary = 1000 if u.username == 'xiaoming' else 2000 for i in range(10): price = base_salary * (1.0 + i / 10.0) i = Item(price=price, category=c2, pub_date='2017-%d-01' % (i + 1), comment='工资') i.save() print("测试数据创建成功")
def handle_uploaded_file_category(f, request): destination = open('upload/csv/name.csv','wb') for chunk in f.chunks(): destination.write(chunk) destination.close() csv_file = open('upload/csv/name.csv','rU') reader = csv.reader(csv_file, dialect='excel') i=0 for line in reader: if i>0: if line[0]==P_CATEGORY_NULL_NAME: data=Category(name=gb_decode(line[1]), isIncome = (line[2]=='True'), user=request.user) data.save() else: pcategory = Category.objects.filter(user__username=request.user.username).filter(name=gb_decode(line[0])) if not pcategory: pass else: data=Category(p_category = pcategory[0], name=gb_decode(line[1]), isIncome = (line[2]=='True'), user=request.user) data.save() i=i+1 csv_file.close() destination = open('upload/csv/name.csv','w') destination.close()
def new_category(request): if request.method == 'POST' and request.is_ajax(): form = CategoryForm(request, data=request.POST.copy()) if form.is_valid(): # 将数据保存到数据库 clean_data = form.cleaned_data # print("clean_data: ", clean_data) # form.add_error('category_name', '该分类已经存在') # form.add_error('category_name', '第二个错误') # print("clean_data: ", clean_data) # print("clean_data: ", form.cleaned_data) # print("form.errors: ", form['category_name'].errors, type(form['category_name'].errors)) # print("退出...") category_name = clean_data['category_name'] # 名称 p_category = None if clean_data[ 'p_category'] == '0' else None # 从数据库获取 # 父类名称 value即为id isIncome = bool(int(clean_data['isIncome'])) # 收入/支出 u = request.user # 用户 c = Category(p_category=p_category, name=category_name, isIncome=isIncome, user=u) try: c.save() # 保存到数据库 except IntegrityError: form.add_error('category_name', '该分类已经存在!') else: print("category保存成功") in_json = json.dumps([True, {}]) # 返回操作状态 return HttpResponse(in_json, content_type='application/json') in_json = json.dumps([False, form.errors]) return HttpResponse(in_json, content_type='application/json') # 表单验证不通过,返回错误信息 else: form = CategoryForm(request) return render(request, "jizhang/new_category.html", { 'form': form, })
def first_login_category(userid): new_category=Category(name=u'工作收入',isIncome=True,user_id=userid) new_category.save() pid = new_category.id sub_category=Category(name=u'工资收入',isIncome=True,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'股票收入',isIncome=True,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'奖金收入',isIncome=True,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'其他收入',isIncome=True,user_id=userid,p_category_id=pid) sub_category.save() new_category=Category(name=u'餐饮',isIncome=False,user_id=userid) new_category.save() pid = new_category.id sub_category=Category(name=u'早餐',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'午餐',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'晚餐',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'饮料水果',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'零食',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() new_category=Category(name=u'交通',isIncome=False,user_id=userid) new_category.save() pid = new_category.id sub_category=Category(name=u'公交地铁',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'加油',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'停车过路',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'汽车保养',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'打的',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() new_category=Category(name=u'购物',isIncome=False,user_id=userid) new_category.save() pid = new_category.id sub_category=Category(name=u'生活用品',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'衣裤鞋帽',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'化妆品',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'首饰手表',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'宝宝用品',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'书籍报刊',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() new_category=Category(name=u'医疗',isIncome=False,user_id=userid) new_category.save() pid = new_category.id sub_category=Category(name=u'看病门诊',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'药店买药',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save() sub_category=Category(name=u'保健品',isIncome=False,user_id=userid,p_category_id=pid) sub_category.save()
def auto_gen_categories(userid): new_category = Category(name=u'工作收入', isIncome=True, user_id=userid) new_category.save() pid = new_category.id sub_category = Category(name=u'工资收入', isIncome=True, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'股票收入', isIncome=True, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'奖金收入', isIncome=True, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'其他收入', isIncome=True, user_id=userid, p_category_id=pid) sub_category.save() new_category = Category(name=u'餐饮', isIncome=False, user_id=userid) new_category.save() pid = new_category.id sub_category = Category(name=u'早餐', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'午餐', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'晚餐', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'饮料水果', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'零食', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() new_category = Category(name=u'交通', isIncome=False, user_id=userid) new_category.save() pid = new_category.id sub_category = Category(name=u'公交地铁', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'加油', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'停车过路', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'汽车保养', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'打的', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() new_category = Category(name=u'购物', isIncome=False, user_id=userid) new_category.save() pid = new_category.id sub_category = Category(name=u'生活用品', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'衣裤鞋帽', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'化妆品', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'首饰手表', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'宝宝用品', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'书籍报刊', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() new_category = Category(name=u'医疗', isIncome=False, user_id=userid) new_category.save() pid = new_category.id sub_category = Category(name=u'看病门诊', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'药店买药', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save() sub_category = Category(name=u'保健品', isIncome=False, user_id=userid, p_category_id=pid) sub_category.save()
def init_data(user): """this is init""" category1 = Category(p_category=None,name="cc",isIncome=True,user=user) category1.save() item1 = Item.objects.create(price=100.00,comment="",category=category1,pub_date="2015-05-24") return item1,category1
def auto_gen_categories(user): #工作收入 new_category=Category(name=u'工作收入',isincome=True,user=user) new_category.save() sub_category=Category(name=u'工资收入',isincome=True,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'奖金收入',isincome=True,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'股票收入',isincome=True,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'其它收入',isincome=True,user=user,p_category=new_category) sub_category.save() #餐饮 new_category=Category(name=u'餐饮',isincome=False,user=user) new_category.save() sub_category=Category(name=u'早餐',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'午餐',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'晚餐',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'水果',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'其它',isincome=False,user=user,p_category=new_category) sub_category.save() #交通 new_category=Category(name=u'交通',isincome=False,user=user) new_category.save() sub_category=Category(name=u'公交',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'地铁',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'的士',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'火车',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'其它',isincome=False,user=user,p_category=new_category) sub_category.save() #购物 new_category=Category(name=u'购物',isincome=False,user=user) new_category.save() sub_category=Category(name=u'服装',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'日用品',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'零食',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'其它',isincome=False,user=user,p_category=new_category) sub_category.save() #医疗 new_category=Category(name=u'医疗',isincome=False,user=user) new_category.save() sub_category=Category(name=u'门诊',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'医院',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'买药',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'保健品',isincome=False,user=user,p_category=new_category) sub_category.save() sub_category=Category(name=u'其它',isincome=False,user=user,p_category=new_category) sub_category.save()