newmediamonitoring/update_newmedia.py

#更新新媒体脚本，结果为有出入的新媒体列表。
'''
1,将excel转为csv，使用utf-8编码集。
2、获取csv内容
3、与数据库中的数据逐项对比
'''
import csv
import uuid

import psycopg2 as psycopg2

# 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!@#'

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 '微信' 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)
                            with connection.cursor() as cursor_p:
                                cursor_p.execute(
                                    "select * from dashboard_area_code_2020 where name = '%s' or name  = '%s' or name = '%s'" % (r[9],r[10],r[11]))
                                connection.commit()
                                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]))
                            #     connection.commit()
                            #     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]))
                            #     connection.commit()
                            #     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()
                                # connection.commit()

                                for c in cursor:
                                    # print(str(c[10]) + "1111111111111111111111")
                                    with connection.cursor() as cursor2:
                                        try:
                                            cursor2.execute(
                                                "select * from dashboard_organization where id = '%s'" % (c[10]))
                                            connection.commit()
                                            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(p_name, c_name, d_name, '444444444444444444444')
                                                if str(name) == str(c2[1]):
                                                    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, 'except'])

                            # try:
                            #
                            # except:
                            #     writer.writerow(
                            #         [r[0], name, code, identificationcode, biz, p_name, c_name,
                            #          d_name,'except'])

if __name__ == '__main__':
    get_csv_weixin('D:/2020/新媒体监测/TASK_ALL.csv')