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