mysql 到postgresql

 1 import pandas as pd
 2 import psycopg2
 3 from io import StringIO
 4 import pymysql
 5 conf={"mysql_form":{
 6             'host' : 'localhost',
 7             'port' : 3306,
 8             'db' : 'pricefluctuation',
 9             'user' : 'root',
10             'passwd' : 'huzu_mysql_QazWsx12#',
11             'charset':'utf8',
12                 },
13         "mysql_test":{
14         'host' : '192.168.2.251',
15         'port' : 3306,
16         'db' : 'pricefluctuation',
17         'user' : 'hewen',
18         'passwd' : 'meiyoumima',
19         'charset':'utf8',
20         },
21         "postgres":{
22         'host' : "120.79.135.9",
23         'port' : "5432",
24         'user' : "postgres",
25         'password' : '33068080',
26
27         },
28     }
29 def sql_to_df():
30     con=pymysql.connect(**conf["mysql_form"])
31     sql="select category_id,name,gameid,img_url from category where category_id in (11665,9653,6504) "
32     try:
33         with con.cursor() as cur:
34             cur.execute(sql)
35             datas=cur.fetchall()
36     finally:
37         con.close()
38     df=pd.DataFrame(list(datas))
39     return df
40
41 def df_to_pg(df=None,db="spider",table='price_730'):
42     #这里生成的StringIO类似于file文件，可read,可write,不同的是file是在硬盘上的文件，而StringIO是在内存中的文件
43     output=StringIO()
44     #也可以直接将字符串转换成内存文件 ，
45     #将数据保存到‘文件’
46     if not df :
47         df=sql_to_df()
48     df.to_csv(output,sep='\t',index=False,header=False)
49     #使文件定位到第一个字符
50     output.seek(0)
51     try:
52         conn=psycopg2.connect(database=db,**conf["postgres"])
53         cur=conn.cursor()
54         cur.copy_from(output,table,null='')
55         conn.commit()
56         result=cur.rowcount
57     finally:
58         cur.close()
59         conn.close()
60     return result
61
62 if __name__=='__main__':
63     # df=pd.read_csv('e:/730_price.csv')
64     print(df_to_pg(table="category"))
65