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 save(self, id=None):
new_category = Category(name=self.cleaned_data['name'],
user=self.user,
p_category=self.cleaned_data['p_category'],
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 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 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 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()
