def select_item(item_id):
if not is_login():
return render_template('login.html')
data = {}
uid = request.cookies.get('the_work_out_uid')
if uid:
item = db_api.search('item', [('id', '=', item_id)])[0]
first_page_href = '/index?uid=' + uid
data['first_page_href'] = first_page_href
data['uid'] = uid
data['item'] = item
part = db_api.search('part', [('id', '=', item['part'])])[0]
data['part'] = part
min, max = date_min_max(datetime.date.today())
records = db_api.search('record', [('item', '=', item_id),
('datetime', '>', min),
('datetime', '<', max)])
if len(records) > 1:
records = make_sort(records)
index = 1
total = 0.0
for r in records:
r['index'] = index
total += r['total']
index += 1
data['record_lists'] = records
data['total'] = total
return render_template('items.html', data=data)
def work_out():
if not is_login():
return render_template('login.html')
data = {}
uid = request.cookies.get('the_work_out_uid')
if uid:
first_page_href = '/index?uid=' + uid
data['first_page_href'] = first_page_href
day_7_ago = datetime.datetime.now() - datetime.timedelta(days=7)
day_7_ago_timestamp = time.mktime(day_7_ago.timetuple())
parts = db_api.search('part', [('user', '=', int(uid))])
all_total = 0.0
for part in parts:
total = 0.0
items = db_api.search('item', [('part', '=', part['ID'])])
for item in items:
records = db_api.search(
'record', [('item', '=', item['ID']),
('datetime', '>', day_7_ago_timestamp)])
for record in records:
total += record['total']
part['total'] = total
all_total += total
data['parts'] = parts
data['all_total'] = all_total
return render_template('work_out.html', data=data)
def delete_item(item_id):
records = db_api.search('record', [('item', '=', item_id)])
for r in records:
db_api.delete('record', r['ID'])
item = db_api.search('item', [('id', '=', item_id)])[0]
db_api.delete('item', item_id)
return select_part(item['part'])
def delete_part(part_id):
items = db_api.search('item', [('part', '=', part_id)])
for i in items:
records = db_api.search('record', [('item', '=', i['ID'])])
for r in records:
db_api.delete('record', r['ID'])
db_api.delete('item', i['ID'])
db_api.delete('part', part_id)
return start_record()
def index():
if not is_login():
return render_template('login.html')
data = request.values.to_dict()
user_id = data['uid']
user = db_api.search('user', [('id', '=', user_id)])[0]
parts = db_api.search('part', [('user', '=', user_id)])
parts_list = []
for part in parts:
parts_list.append(part)
return work_out()
def select_part(part_id):
if not is_login():
return render_template('login.html')
data = {}
uid = request.cookies.get('the_work_out_uid')
if uid:
part = db_api.search('part', [('id', '=', part_id)])[0]
first_page_href = '/index?uid=' + uid
data['first_page_href'] = first_page_href
data['uid'] = uid
data['part'] = part
items = db_api.search('item', [('part', '=', part_id)])
data['item_lists'] = items
return render_template('parts.html', data=data)
def login():
if request.method == 'POST':
datax = request.form.to_dict()
user = db_api.search('user', [('name', '=', datax['name']),
('password', '=', datax['password'])])
if not user:
content = json.dumps({
"success": False,
'data': {
"reason": '用户名密码错误。'
}
})
return Response(content)
cookie = str(random.randint(1111111111, 9999999999))
db_api.update('user', user[0]['ID'], {'cookie': cookie})
content = json.dumps({
"success": True,
'data': {
"uid": user[0]['ID'],
'cookie': cookie
}
})
return Response(content)
elif request.method == 'GET':
return render_template('login.html')
def is_login():
uid = request.cookies.get('the_work_out_uid')
cookie = request.cookies.get('the_work_out_cookie')
if uid and cookie:
if not db_api.search('user', [('ID', '=', int(uid)),
('cookie', '=', cookie)]):
return False
else:
return False
return True
def start_record():
if not is_login():
return render_template('login.html')
data = {}
uid = request.cookies.get('the_work_out_uid')
if uid:
first_page_href = '/index?uid=' + uid
data['first_page_href'] = first_page_href
data['uid'] = uid
parts = db_api.search('part', [('user', '=', uid)])
data['parts'] = parts
return render_template('records.html', data=data)
def find_record():
uid = request.cookies.get('the_work_out_uid')
datax = request.values.to_dict()
date_str = datax['date_str']
if date_str == 'today':
date = datetime.datetime.now().date()
else:
if datax.get('day') == 'yesterday':
date = (datetime.datetime.strptime(date_str, '%Y-%m-%d') -
datetime.timedelta(days=1)).date()
elif datax.get('day') == 'tomorrow':
date = (datetime.datetime.strptime(date_str, '%Y-%m-%d') +
datetime.timedelta(days=1)).date()
else:
date = datetime.datetime.strptime(date_str, '%Y-%m-%d').date()
min, max = date_min_max(date)
parts = db_api.search('part', [('user', '=', uid)])
all_total = 0.0
for part in parts:
records_list = []
total = 0.0
items = db_api.search('item', [('part', '=', part['ID'])])
for item in items:
records = db_api.search('record', [('item', '=', item['ID']),
('datetime', '>', min),
('datetime', '<', max)])
for record in records:
total += record['total']
record['item_name'] = item['name']
records_list += records
part['records_list'] = records_list
part['total'] = total
all_total += total
content = json.dumps({
"success": True,
'data': parts,
'total': all_total,
'today': str(date)
})
return Response(content)
def signup():
if request.method == 'POST':
datax = request.values.to_dict()
user = db_api.search('user', [('name', '=', datax['name'])])
if user:
content = json.dumps({
"success": False,
'data': {
"reason": '该用户名已经被注册了。'
}
})
return Response(content)
else:
cookie = str(random.randint(1111111111, 9999999999))
user_id = db_api.insert(
'user', {
'name': datax['name'],
'password': datax['password'],
'cookie': cookie
})
content = json.dumps({
'success': True,
'data': {
'uid': user_id,
'cookie': cookie
}
})
return Response(content)
else:
content = json.dumps({
"success": False,
'data': {
"reason": '不是post方法提交'
}
})
resp = Response(content)
return resp
def delete_record(record_id):
record = db_api.search('record', [('id', '=', record_id)])[0]
db_api.delete('record', record_id)
return select_item(record['item'])
def make_sort(ret):
new_ret = []
for i in ret:
seq = 0
for j in new_ret:
if i['date'] < j['date']:
break
seq += 1
new_ret.insert(seq, i)
return new_ret
if __name__ == '__main__':
result = crawl()
done = db_api.search('instagram_teddy')
done_date = [j['date'] for j in done]
ret = [i for i in result if i['date'] not in done_date] # 将要写入的动态
# 对ret中的元素按date排序
if len(ret) > 1:
ret = make_sort(ret)
print '', datetime.datetime.now(), ret
for j in ret:
# 下载图片并写入
ss = requests.session()
video = ss.get(j['video']).content if 'video' in j.keys() else None
pic = ss.get(j['pic']).content
txt = j['txt'] + '\r\n\r\n' + u"teddysphotos' instagram\r\n" + str(
datetime.datetime.fromtimestamp(j['date']))
result = same.same_api(pic, txt, video)
# 成功写入same