Пример #1
0
class DbUtil(object):
    LOG = LogUtil.getLogger()

    def __init__(self):
        self.LOG.info("数据库操作工具类初始化完成")

    def getConnect(self):
        self.LOG.info("获取数据库连接")
        try:
            return pymysql.connect(host=Config.mysql_host, port=Config.mysql_port, user=Config.mysql_user,
                                   passwd=Config.mysql_passwd, db=Config.mysql_db, charset=Config.mysql_charset)
        except Exception as e:
            error = '数据库连接异常! ERROR : %s' % (e)
            self.LOG.error(error)
            return None

    def dbClose(self, db):
        self.LOG.info("关闭数据库连接")
        if db:
            db.close()

    def _query(self, sql=''):
        self.LOG.info("执行查询语句,SQL:" + sql)
        db = self.getConnect()
        try:
            cursor = db.cursor()
            # 执行SQL语句
            cursor.execute(sql)
            # 获取所有记录列表
            results = cursor.fetchall()
            return results
        except Exception as e:
            error = '数据查询异常! ERROR : %s' % (e)
            self.LOG.error(error)
        finally:
            self.dbClose(db)
Пример #2
0
#!/usr/bin/python

import datetime
import os
import sys

base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(base_dir)

import util.DbUtil as DbUtil
import util.LogUtil as LogUtil

dbUtil = DbUtil.DbUtil()
LOG = LogUtil.getLogger()


# 广告点击记录每日汇总
def day():
    LOG.info("广告点击记录每日汇总开始=============>")
    db = dbUtil.getConnect()
    try:
        cursor = db.cursor()
        sql = '''SELECT DATE_FORMAT(NOW(),'%Y-%m-%d %H')'''
        cursor.execute(sql)
        now_str = cursor.fetchone()[0]
        now = datetime.datetime.strptime(now_str, '%Y-%m-%d %H')
        count = 1
        if now_str[11:13] == '00':
            now = (now + datetime.timedelta(days=-2))
            count = 2
        else: