diff --git a/monitor/views.py b/monitor/views.py index f17e793..4073ac2 100644 --- a/monitor/views.py +++ b/monitor/views.py @@ -131,8 +131,13 @@ def timeliness_monitoring_weixin(request): this_week_end = now + timedelta(days=6 - now.weekday()) # 本月第一天和最后一天 this_month_start = datetime.datetime(now.year, now.month, 1) - this_month_end = datetime.datetime(now.year, now.month + 1, 1) - timedelta(days=1) + datetime.timedelta( - hours=23, minutes=59, seconds=59) + if now.month == 12: + this_month_end = datetime.datetime(now.year, now.month , 1) - timedelta(days=1) + datetime.timedelta( + hours=23, minutes=59, seconds=59) + else: + this_month_end = datetime.datetime(now.year, now.month, 1) - timedelta(days=1) + datetime.timedelta( + hours=23, minutes=59, seconds=59) + # 新媒体数量 new_media_count = int(Weixin.objects.all().count()) + int(Weibo.objects.all().count()) + int( Toutiao.objects.all().count()) + int(Douyin.objects.all().count()) + int(Qita.objects.all().count()) diff --git a/update_newmedia.py b/update_newmedia.py index 93d45b4..a4e310c 100644 --- a/update_newmedia.py +++ b/update_newmedia.py @@ -5,14 +5,95 @@ 3、与数据库中的数据逐项对比 ''' import csv +import uuid +import psycopg2 as psycopg2 -def get_csv(path): - with open(path, encoding='utf-8') as csvfile: - reader = csv.reader(csvfile) - for r in reader: +# code = None +# alias = None +# attention = None +# remark = None +# identificationcode = None +# function = None +# articleurl = None +# weixinid = None +# type = None +G2 = 'host=210.77.68.250 port=5432 dbname=newmediaDB3 user=newmedia password=newmedia2020!@#' - print(r) +def get_csv_weixin(path): + with open("D:/2020/新媒体监测/新媒体对比结果_其他.csv", "w", newline='') as csvfile: + writer = csv.writer(csvfile) + writer.writerow( + ["序号", "冲突账号"]) + with psycopg2.connect(G2) as connection: + with open(path, encoding='utf-8') as csvfile: + reader = csv.reader(csvfile) + for r in reader: + if r[1] != '单位全称': + if '头条' not in r[7] and '微信' not in r[7] and '微博' not in r[7] and '抖音' not in r[7]: + name = r[1] + code = r[4] + identificationcode = r[3] + function = r[8] + articleurl = r[12] + biz = r[13] + province = None + cities = None + district = None + d_name = None + c_name = None + p_name = None + + print(r) + try: + with connection.cursor() as cursor_p: + cursor_p.execute( + "select * from dashboard_area_code_2020 where name = '%s'" % (r[9])) + for p in cursor_p: + province = p[1] + p_name = p[2] + with connection.cursor() as cursor_c: + cursor_c.execute( + "select * from dashboard_area_code_2020 where name = '%s'" % (r[10])) + for c in cursor_c: + cities = c[1] + c_name = c[2] + with connection.cursor() as cursor_d: + cursor_d.execute( + "select * from dashboard_area_code_2020 where name = '%s'" % (r[11])) + for d in cursor_d: + district = d[1] + d_name = d[2] + print(code) + with connection.cursor() as cursor: + cursor.execute( + "select * from dashboard_qita where code = '%s'" % (code)) + # connection.commit() + + for c in cursor: + print(str(c[10]) + "1111111111111111111111") + with connection.cursor() as cursor2: + cursor2.execute( + "select * from dashboard_organization where id = '%s'" % (c[10])) + for c2 in cursor2: + print(str(c2[0]) + "55555555555555555555555") + + if str(name) == str(c2[1]) and str(province) == str(c2[3]) and str( + cities) == str(c2[4]) and str(district) == str(c2[5]) and str( + code) == str(c[1]) and str(identificationcode) == str( + c[13]) and str(biz) == str(c[9]): + print(c2) + else: + writer.writerow( + [r[0], name, code, identificationcode, biz, p_name, c_name, + d_name]) + writer.writerow( + [r[0], c2[1], c[1], c[13], c[9], c2[3], c2[4], c2[5]]) + + except: + writer.writerow( + [r[0], name, code, identificationcode, biz, p_name, c_name, + d_name]) if __name__ == '__main__': - get_csv('D:/2020/新媒体监测/TASK_ALL.csv') \ No newline at end of file + get_csv_weixin('D:/2020/新媒体监测/TASK_ALL.csv') \ No newline at end of file