utils.py 874 Bytes
Newer Older
高雅喆's avatar
高雅喆 committed
1 2 3 4 5 6 7 8 9 10
# -*- coding: UTF-8 -*-
import pymysql
import datetime

def con_sql(sql):
	#从数据库的表里获取数据
	"""
	:type sql : str
	:rtype : tuple
	"""
11
	db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_prod')
高雅喆's avatar
高雅喆 committed
12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
	cursor = db.cursor()
	cursor.execute(sql)
	result = cursor.fetchall()
	db.close()
	return result

def tuple2dict(tuple_result):
	#把sql结果从tuple(tuple,tuple)格式转换成dict格式
	"""
	:type tuple_result : tuple
	:rtype : dict
	"""
	dict_result = {}
	for i in range(len(tuple_result)):
		dict_result[tuple_result[i][0]] = tuple_result[i][1]
	return dict_result

def get_yesterday_date():
30
	#自动获取昨天的日期,如"2018-08-08"
高雅喆's avatar
高雅喆 committed
31 32 33 34
	"""
	:rtype : str
	"""
	today = datetime.date.today()
高雅喆's avatar
高雅喆 committed
35
	yesterday = today - datetime.timedelta(days=1)
36
	yesterday = yesterday.strftime("%Y-%m-%d")
高雅喆's avatar
高雅喆 committed
37
	return yesterday