save

test.py

@@ -1,16 +1,31 @@
-import json
+# 连接数据库
+import psycopg2
+from psycopg2.extras import DictCursor
+import logging
 
-arg="{\n  \"img1\": \"predict/upload/8afb5a7b-82a7-4654-96e7-c32b7fab857clogo-site.png\",\n  \"img2\": \"predict/upload/b27f1346-097d-47c1-9d26-877c66aca42dlogo-site-dark.png\"\n}\n"
+# 配置日志
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 
-def main(arg1):
-   # 解析 JSON 字符串
-    data_dict = json.loads(arg1)
+conn = psycopg2.connect(
+    dbname="real3d",
+    user="postgres",
+    password="postgis",
+    host="192.168.1.177",
+    port="5432"
+)
 
-    # 获取并输出 data 字段内容
-    img1 = data_dict.get("img1", "")
-    img2 = data_dict.get("img2", "")
-    print({"img1":img1,"img2":img2})
-    return {"img1":img1,"img2":img2}
 
-main(arg)
+def getDiffEwkt():
+    with conn.cursor(cursor_factory=DictCursor) as cur:
+        sql = f"""SELECT * FROM	base."t_fzss_fzxz" limit 1;"""
+        complete_sql = cur.mogrify(sql, ).decode('utf-8')
+        logger.info(f"Executing SQL: {complete_sql}")
+        cur.execute(sql, )
+        res = cur.fetchone()
+        print(res)
+    return res[0]
+
+
+getDiffEwkt()

tif_merge.py

@@ -0,0 +1,41 @@
+import rasterio
+from rasterio.merge import merge
+import os
+
+inPath = '../../data/tif_crop/'
+
+parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), inPath))
+tif_files = os.listdir(parent_dir)
+
+# 读取所有 TIF 文件并放入列表
+src_files_to_mosaic = []
+for tif in tif_files:
+    src = rasterio.open(parent_dir + '/' + tif)
+    src_files_to_mosaic.append(src)
+
+# 拼接 TIF 文件
+mosaic, out_trans = merge(src_files_to_mosaic)
+
+# 获取拼接后的元数据
+out_meta = src_files_to_mosaic[0].meta.copy()
+
+# 更新拼接后文件的元数据
+out_meta.update({
+    "driver": "GTiff",
+    "count": mosaic.shape[0],  # 波段数
+    "crs": src_files_to_mosaic[0].crs,  # 坐标参考系统
+    "transform": out_trans,  # 转换矩阵
+    "width": mosaic.shape[2],  # 拼接后的图像宽度
+    "height": mosaic.shape[1]  # 拼接后的图像高度
+})
+
+# 保存拼接后的结果为新文件
+outPath = '/home/project/paddlers/data/tif_merge/tif_mosaic.tif'
+with rasterio.open(outPath, "w", **out_meta) as dest:
+    dest.write(mosaic)
+
+# 关闭所有文件
+for src in src_files_to_mosaic:
+    src.close()
+
+print(f"拼接后的文件保存为: {outPath}")