pg2gdb.py 1.7 KB
# coding=utf-8
#author:        4N
#createtime:    2021/10/21
#email:         nheweijun@sina.com

from osgeo import ogr
from osgeo import gdal

def function():
    gdal.SetConfigOption("SHAPE_ENCODING", "UTF-8")

    fn = "PG: user=postgres password=chinadci host=172.26.60.101 port=5432 dbname=parcel"
    driver = ogr.GetDriverByName("PostgreSQL")
    ds:ogr.DataSource = driver.Open(fn, 1)

    layer:ogr.Layer = ds.GetLayerByName("ZDCLHX2000V14")

    output = r"E:\D2\桌面\工作\2021.10省厅测评\10w.shp"

    shp_driver = ogr.GetDriverByName("FileGDB")
    output_ds: ogr.DataSource = shp_driver.CreateDataSource(output)

    out_layer:ogr.Layer = output_ds.CreateLayer(layer.GetName(),layer.GetSpatialRef(),layer.GetGeomType())

    field_names = ["id","预编号","土地使用者","所有权性质","土地证号","地址","镇区","测量日期","二维码信息","shape_area"]
    field_names_yinwen = ["id", "precode", "user", "xingzhi", "tudi", "address", "region", "date", "code", "shape_area"]
    scheme = [field for field in layer.schema if field.GetName() in field_names]

    out_layer.CreateFields(scheme)
    count = 100000

    i=0
    featureDefn = out_layer.GetLayerDefn()

    for f in layer:
        ff :ogr.Feature = f
        new_f = ogr.Feature(featureDefn)
        new_f.SetGeometry(ff.geometry())
        for field in field_names:
            dd = ff.GetFieldIndex(field)
            kkk = ff.GetField(dd)
            new_f.SetField(field,kkk)
        out_layer.CreateFeature(new_f)
        i += 1
        if i==count:
            break

    output_ds.Destroy()
if __name__ == '__main__':

    function()