69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
from ops.plog import define_logger
|
|
import logging
|
|
import sys
|
|
import datetime
|
|
from ops.mmysql import MysqlBase
|
|
import pdb
|
|
|
|
define_logger("/data/logs/get_newusers_days.log")
|
|
log = logging.getLogger(__name__)
|
|
|
|
DB = {'user': 'mytga', 'pswd': 'gzVwh4HGR68G', 'host': '10.10.3.5', 'db': 'ad'}
|
|
|
|
mydb = MysqlBase(**DB)
|
|
table_name = "new_users_days"
|
|
|
|
|
|
def get_data(times, gameid, channelid):
|
|
sql = f"""SELECT
|
|
count(accountid) AS users,
|
|
ad_channel
|
|
FROM
|
|
newuser
|
|
WHERE
|
|
gameid={gameid}
|
|
AND channelid={channelid}
|
|
AND date(register_time) ='{times}'
|
|
GROUP BY
|
|
ad_channel;"""
|
|
out = mydb.query(sql)
|
|
data = []
|
|
if out:
|
|
try:
|
|
for line in out:
|
|
temp = {}
|
|
temp['gameid'] = gameid
|
|
temp['channelid'] = channelid
|
|
temp['usernum'], temp['ad_channel'] = line
|
|
data.append(temp)
|
|
except Exception:
|
|
log.error(f"split {line} error!")
|
|
return data
|
|
|
|
|
|
def write2db(data, times, gameid, channelid):
|
|
for line in data:
|
|
line['date'] = times
|
|
line['key'] = f"{gameid}#{channelid}#{line['ad_channel']}#{times}"
|
|
try:
|
|
mydb.insert(table_name, line)
|
|
except Exception:
|
|
log.error(f" insert {line} to db failed", exc_info=True)
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) == 2:
|
|
times = sys.argv[1]
|
|
else:
|
|
times = (datetime.date.today() - datetime.timedelta(days=1)).strftime('%Y-%m-%d')
|
|
gameids = (1004, 1011, 2001, 1001, 1013)
|
|
for gameid in gameids:
|
|
channelid = 6001
|
|
data = get_data(times, gameid, channelid)
|
|
write2db(data, times, gameid, channelid)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|