/
views.py
457 lines (385 loc) · 13.6 KB
/
views.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
from flask import render_template
from main import app
from flask import redirect
from flask import session
import hashlib
import datetime
import functools
from models import *
class Pager:
"""
flask分页通过sqlalachemy查询进行分页
offset 偏移,开始查询的位置
limit 单页条数
分页器需要具备的功能
页码
分页数据
是否第一页
是否最后一页
"""
def __init__(self, data, page_size):
"""
:param data: 要分页的数据
:param page_size: 每页多少条
"""
self.data = data # 总数据
self.page_size = page_size # 单页数据
self.is_start = False
self.is_end = False
self.page_count = len(data)
self.next_page = 0 # 下一页
self.previous_page = 0 # 上一页
self.page_nmuber = self.page_count / page_size
# (data+page_size-1)//page_size
if self.page_nmuber == int(self.page_nmuber):
self.page_nmuber = int(self.page_nmuber)
else:
self.page_nmuber = int(self.page_nmuber) + 1
self.page_range = range(1, self.page_nmuber + 1) # 页码范围
def page_data(self, page):
"""
返回分页数据
:param page: 页码
page_size = 10
1 offect 0 limit(10)
2 offect 10 limit(10)
page_size = 10
1 start 0 end 10
2 start 10 end 20
3 start 20 end 30
"""
self.next_page = int(page) + 1
self.previous_page = int(page) - 1
if page <= self.page_range[-1]:
page_start = (page - 1) * self.page_size
page_end = page * self.page_size
# data = self.data.offset(page_start).limit(self.page_size)
data = self.data[page_start:page_end]
if page == 1:
self.is_start = True
else:
self.is_start = False
if page == self.page_range[-1]:
self.is_end = True
else:
self.is_end = False
else:
data = ["没有数据"]
return data
def setPassword(password):
result = hashlib.md5(password.encode()).hexdigest()
return result
class Calendar:
"""
当前类实现日历功能
1、返回列表嵌套列表的日历
2、安装日历格式打印日历
# 如果一号周周一那么第一行1-7号 0
# 如果一号周周二那么第一行empty*1+1-6号 1
# 如果一号周周三那么第一行empty*2+1-5号 2
# 如果一号周周四那么第一行empty*3+1-4号 3
# 如果一号周周五那么第一行empyt*4+1-3号 4
# 如果一号周周六那么第一行empty*5+1-2号 5
# 如果一号周日那么第一行empty*6+1号 6
# 输入 1月
# 得到1月1号是周几
# [] 填充7个元素 索引0对应周一
# 返回列表
# day_range 1-30
"""
def __init__(self, month="now"):
self.result = []
big_month = [1, 3, 5, 7, 8, 10, 12]
small_month = [4, 6, 9, 11]
# 获取当前月
now = datetime.datetime.now()
if month == "now":
month = now.month
first_date = datetime.datetime(now.year, now.month, 1, 0, 0)
# 年 月 日 时 分
else:
# assert int(month) in range(1,13)
first_date = datetime.datetime(now.year, month, 1, 0, 0)
if month in big_month:
day_range = range(1, 32) # 指定月份的总天数
elif month in small_month:
day_range = range(1, 31)
else:
day_range = range(1, 29)
# 获取指定月天数
self.day_range = list(day_range)
first_week = first_date.weekday() # 获取指定月1号是周几 6
line1 = [] # 第一行数据
for e in range(first_week):
line1.append("empty")
for d in range(7 - first_week):
line1.append(
str(self.day_range.pop(0)) + "—django开发"
)
self.result.append(line1)
while self.day_range: # 如果总天数列表有值,就接着循环
line = [] # 每个子列表
for i in range(7):
if len(line) < 7 and self.day_range:
line.append(str(self.day_range.pop(0)) + "—django开发")
else:
line.append("empty")
self.result.append(line)
def return_month(self):
"""
返回列表嵌套列表的日历
"""
return self.result
def print_month(self):
"""
安装日历格式打印日历
"""
print("星期一 星期二 星期三 星期四 星期五 星期六 星期日")
for line in self.result:
for day in line:
day = day.center(6)
print(day, end=" ")
print()
def loginValid(fun):
@functools.wraps(fun) # 保留原函数的名称
def inner(*args, **kwargs):
username = request.cookies.get("username")
id = request.cookies.get("id", "0")
user = User.query.get(int(id))
session_username = session.get("username")
if user: # 检测是否有对应id的用户
if user.user_name == username and username == session_username: # 用户名是否对应
return fun(*args, **kwargs)
else:
return redirect("/login/")
else:
return redirect("/login/")
return inner
@app.route("/") # 然后再进行路由
@loginValid # 先执行loginValid
def index():
name = "laobian"
return render_template("index.html", **locals())
@app.route("/login/", methods=["get", "post"])
# @csrf.exempt
def login():
error = ""
if request.method == "POST":
form_data = request.form
email = form_data.get("email")
password = form_data.get("password")
user = User.query.filter_by(email=email).first()
if user:
db_password = user.password
if setPassword(password) == db_password:
response = redirect("/index/")
response.set_cookie("username", user.user_name)
response.set_cookie("email", user.email)
response.set_cookie("id", str(user.id))
session["username"] = user.user_name
return response
else:
error = "密码错误"
else:
error = "用户名不存在"
return render_template("login.html", error=error)
@app.route("/logout/")
def logout():
response = redirect("/login/")
response.delete_cookie("username")
response.delete_cookie("email")
response.delete_cookie("id")
session.pop("username")
# del session["username"]
return response
@app.route("/index/") # 官方的路由匹配器
@loginValid
def exindex():
# c = Curriculum()
# c.c_id = "0001"
# c.c_name = "python基础"
# c.c_time = datetime.datetime.now()
# c.save()
curr_list = Curriculum.query.all()
return render_template("ex_index.html", curr_list=curr_list)
@app.route("/userinfo/")
def userinfo():
calendar = Calendar().return_month()
now = datetime.datetime.now()
return render_template("userinfo.html", **locals())
from flask import request
from models import User
@app.route("/register/", methods=["GET", "POST"])
def register():
"""
form表单提交的数据由request.form接受
"""
if request.method == "POST":
username = request.form.get("username")
password = request.form.get("password")
email = request.form.get("email")
user = User()
user.user_name = username
user.password = setPassword(password)
user.email = email
user.save()
return render_template("register.html")
@app.route("/holiday_leave/", methods=["GET", "POST"])
# @csrf.exempt
def holiday_leave():
if request.method == "POST":
data = request.form
request_user = data.get("request_user")
request_type = data.get("request_type")
start_time = data.get("start_time")
end_time = data.get("end_time")
phone = data.get("phone")
request_description = data.get("request_description")
leave = Leave()
leave.request_id = request.cookies.get("id")
leave.request_name = request_user
leave.request_type = request_type # 假期类型
leave.request_start_time = start_time # 起始时间
leave.request_end_time = end_time # 结束时间
leave.request_description = request_description # 请假事由
leave.request_phone = phone # 联系方式
leave.request_status = "0" # 假条状态
leave.save()
return redirect("/leave_list/1/")
return render_template("holiday_leave.html")
@app.route("/leave_list/<int:page>/")
@loginValid
def leave_list(page):
leaves = Leave.query.all()
pager = Pager(leaves, 2)
page_data = pager.page_data(page)
return render_template("leave_list.html", **locals())
import json
from flask import jsonify # flask封装后的json 方法
# @app.route("/cancel/<int:id>/")
# def cancel(id):
# leave = Leave.query.get(id)
# leave.delete()
# return jsonify({"data":"删除成功"})
@app.route("/cancel/", methods=["GET", "POST"])
def cancel():
# data1 = request.args
# data2 = request.data
# data3 = request.form
id = request.form.get("id") # 通过args接受get请求数据
leave = Leave.query.get(int(id))
leave.delete()
return jsonify({"data": "删除成功"}) # 返回json数据
from forms import TaskForm
@app.route("/add_task/", methods=["GET", "POST"])
def add_task():
"""
print(task.errors) 表单校验错误
print(task.validate_on_submit()) 判断是否是一个有效的post请求
print(task.validate()) 判断是否是一个合法的post请求
print(task.data) 提交的数据
"""
errors = ""
task = TaskForm()
if request.method == "POST":
if task.validate_on_submit(): # 判断是否是一个有效的post请求
formData = task.data
else:
errors_list = list(task.errors.keys())
errors = task.errors
print(errors)
return render_template("add_task.html", **locals())
from settings import STATICFILES_DIR
import os
from settings import STATICFILES_DIR
@app.route("/picture/", methods=["GET", "POST"])
def picture():
p = {"picture": "img/1.jpg"}
if request.method == "POST":
file = request.files.get("photo")
file_name = file.filename
file_path = "img/%s" % file_name
save_path = os.path.join(STATICFILES_DIR, file_path)
file.save(save_path)
p = Picture()
p.picture = file_path
p.save()
return render_template("picture.html", p=p)
from main import api
from flask_restful import Resource
@api.resource("/Api/leave/")
class LeaveApi(Resource):
def __init__(self):
"""
定义返回的格式
"""
super(LeaveApi, self).__init__()
self.result = {
"version": "1.0",
"data": ""
}
def set_data(self, leave):
"""
定义返回的数据
"""
result_data = {
"request_name": leave.request_name,
"request_type": leave.request_type,
"request_start_time": leave.request_start_time,
"request_end_time": leave.request_end_time,
"request_description": leave.request_description,
"request_phone": leave.request_phone,
}
return result_data
def get(self):
"""
处理get请求
"""
data = request.args # 获取请求的数据
id = data.get("id") # 获取id
if id: # 如果id存在,返回当前id数据
leave = Leave.query.get(int(id))
result_data = self.set_data(leave)
else: # 如果id不存在,返回所有数据
leaves = Leave.query.all()
result_data = []
for leave in leaves:
result_data.append(self.set_data(leave))
self.result["data"] = result_data
return self.result
def post(self):
"""
这是post请求,负责保存数据
"""
data = request.form
request_id = data.get("request_id")
request_name = data.get("request_name")
request_type = data.get("request_type")
request_start_time = data.get("request_start_time")
request_end_time = data.get("request_end_time")
request_description = data.get("request_description")
request_phone = data.get("request_phone")
leave = Leave()
leave.request_id = request_id
leave.request_name = request_name
leave.request_type = request_type # 假期类型
leave.request_start_time = request_start_time # 起始时间
leave.request_end_time = request_end_time # 结束时间
leave.request_description = request_description # 请假事由
leave.request_phone = request_phone # 联系方式
leave.request_status = "0" # 假条状态
leave.save()
self.result["data"] = self.set_data(leave)
return self.result
def put(self):
data = request.form
id = data.get("id")
print(data)
print(data.keys())
leave = Leave.query.get(int(id))
leave.save()
self.result["data"] = self.set_data(leave)
return self.result
def delete(self):
data = request.form
return {"method": "这是delet请求,负责删除数据"}