读取.bcp文件写入postgre数据库

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/weixin_44266650/article/details/98652782

# -*- coding: utf-8 -*-
from Config import config
from Config.config import file_path
# from Config.config import project_config
import psycopg2
import os
import datetime


# file_name_path = os.path.join(file_path)
# f = open(file_name_path, "r", encoding="utf-8")
# for i in f:
#     a = i.split('\t')
#     print(a)


#导入数据
def pg_source():
    #连接
    conn_full = psycopg2.connect(**config.pg_config_full)
    try:

        file_name_path = os.path.join(file_path)
        f = open(file_name_path, "r", encoding="utf-8")
        for i in f:
            # 对数据进行处理
            pg_s = i.split('\t')
            #判断数据是否准确
            if len(pg_s) != 17:
                print('error %s' % pg_s)
            cur = conn_full.cursor()
            sql = "insert into public.t_ori_mac_data (netbar_wacode,collection_equipmentid,mac,brand,capture_time,terminal_fieldstrength,fn) values('%s','%s','%s','%s','%s','%s','%s') " % (pg_s[13],pg_s[14],pg_s[0],pg_s[1],pg_s[3],pg_s[4],file_path.split('/')[-1])
            cur.execute(sql)
            conn_full.commit()
            
    except Exception as e:
        print(e)

if __name__ == "__main__":
    pg_source()