将pgadmin导出的数据转换成INSERT语句

最新推荐文章于 2022-02-09 11:43:17 发布

转载最新推荐文章于 2022-02-09 11:43:17 发布 · 504 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Mingxx/archive/2013/04/25/3041490.html

文章标签：

#数据库 #python

本文介绍了一个使用Python编写的简单脚本，该脚本能够将指定格式的文本文件转换为SQL插入语句。通过解析命令行参数指定的表名，并读取文本文件中的数据，该脚本可以方便地生成用于数据库导入的SQL文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#!/usr/bin/python

import os
import sys
import re
import getopt

_SPLITE_ = ';'

table_name = ''

def parserFile(path):
    try:
        f = open(path, 'r')
    except:
        print "open [%s] failed." % path
        return

    lines = f.readlines()

    f.close()
    
    if len(lines) <= 0:
        print "no data in file"

    headers = lines[0].strip("\n").split(_SPLITE_)
    lines.pop(0)

    datas = []
    items = len(headers)
    for l in lines:
        tmp = l.strip('\n').split(_SPLITE_)
        datas.append(tmp)

    try:
        f = open(path + ".sql", 'w')
    except:
        print "open [%s] failed." % path
        return
   
    for data in datas:
        insert_string_header = "INSERT %s(%s) VALUES(%s);\n" % (table_name, ','.join(headers), ','.join(data))
        f.write(insert_string_header)

    f.close()


def main(argv):
    global table_name
    try: 
        opts, args = getopt.getopt(argv, 't:',['table-name='])
    except getopt.GetoptError:
        print "paramter error"
        return

    for o, a in opts:
        if o in ('-t', '--table-name'):
            table_name = a
            print a
        else:
            print "unknown paramter %s" % a
    
    for a in args:
        parserFile(a)

if __name__ == '__main__':
    main(sys.argv[1:])