55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
# 更新新媒体脚本，结果为有出入的新媒体列表。
|
||
'''
|
||
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 set_csv_weixin(path):
|
||
with open(path, encoding='utf-8') as csvfile:
|
||
reader = csv.reader(csvfile)
|
||
for r in reader:
|
||
if r[1] == '删除':
|
||
print(r)
|
||
# with psycopg2.connect(G2) as connection:
|
||
# with connection.cursor() as cursor:
|
||
# cursor.execute(
|
||
# "delete from dashboard_organization where name = '%s'" %(r[0]))
|
||
# connection.commit()
|
||
elif r[1] == '新增':
|
||
with open('D:/2020/新媒体监测/TASK_ALL.csv',encoding='utf-8') as c:
|
||
t = csv.reader(c)
|
||
for i in t:
|
||
if r[0].replace(' ', '') == i[1].replace(' ', ''):
|
||
print(i[1], r[0])
|
||
with psycopg2.connect(G2) as connection:
|
||
with connection.cursor() as cursor:
|
||
|
||
id = uuid.uuid4()
|
||
cursor.execute(
|
||
"insert into dashboard_organization(id,name,image, province, cities, district, town,village,directly,status,level_id,id_code, created, updated) values (%s,%s, 'danweimoren.jpg','620000000000',%s,%s,'','', %s,'1','69be9ef4-b7b7-4049-a86e-7083bee40f0e',%s,now(),now())"
|
||
, (str(id), i[1],i[10],i[11],i[6],i[2]))
|
||
connection.commit()
|
||
|
||
|
||
|
||
|
||
if __name__ == '__main__':
|
||
set_csv_weixin('D:/2020/新媒体监测/单位名称对比结果.csv')
|