python将本地png切片栅格数据写入postgis（Postgre入门三）

1、准备数据

2、配置好postgre

3、python写入

import psycopg2 as ps
import os
import glob

#找到文件路径下的所有表格名称,返回列表
def getallpngs(file_dir):
     file_lists=[]
     for i,j,k in os.walk(file_dir):
          file_list=glob.glob(os.path.join(i,"*.png"))
          file_lists=file_lists+file_list

     return file_lists

# 连接数据库
def getconn():
     return ps.connect(database=database,user=user,password=password,host=host,port=port)

# 创建数据表
def create_table(tablename,sql):
     conn = getconn()
     cur = conn.cursor()
     try:
          cur.execute(sql)
          conn.commit()
          cur.close()
          conn.close()
          print("{}数据库创建成功！！！".format(tablename))
     except Exception as e:
          print("{}数据库表已存在！！！".format(tablename))
# 插入数据
def insert_data(tablename,data):
     conn = getconn()
     cur = conn.cursor()
     key_list = []
     value_list=[]
     sql = '''insert into {}(%s) values(%s);'''.format(tablename)
     
     for k,v in data.items():
          if k!='id' and k!='tablename':
               key_list.append(k)
               value_list.append('%%(%s)s' % k)
     sql = sql % (','.join(key_list),','.join(value_list))
     
     try:
          cur.execute(sql,data)
          conn.commit()
          # print('插入数据成功')
     except:
          conn.rollback()
          # print("插入数据失败")
     cur.close()
     conn.close()

# 查询表数据
def query(sql):
     conn= getconn()
     cursor = conn.cursor()
     cursor.execute(sql)
     # 返回所有数据
     class_list = cursor.fetchall()
     cursor.close()
     conn.close()
     return class_list
if __name__ == "__main__":
     tablename="img_test"
     sql = "create table {} (name text NOT NULL, content bytea NOT NULL);".format(tablename)
     # 1、创建数据库表
     create_table("table_tdt",sql)
     
     # 2、获取本地数据路径
     image_dir = './data' #the path of images
     file_lists=getallpngs(image_dir)
     
     # 3、写入数据
     for fileone in file_lists:
          with open(fileone,'rb') as reader:
               buffer = reader.read()     
          data={}
          data['name']=fileone
          data['content']=ps.Binary(buffer)
          insert_data(tablename,data)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

4、数据库中是这样的