init

import_osm.py

@@ -0,0 +1,58 @@
+import json
+import psycopg2
+from shapely.geometry import shape
+
+def import_geojson(filepath):
+    conn = psycopg2.connect(
+        dbname="gis_data",
+        user="osm",
+        password="osm",
+        host="localhost"
+    )
+    cur = conn.cursor()
+
+    try:
+        with open(filepath) as f:
+            data = json.load(f)
+
+        if data["type"] != "FeatureCollection":
+            raise ValueError("GeoJSON must be a FeatureCollection")
+
+        for idx, feature in enumerate(data["features"]):
+            try:
+                geom_type = feature["geometry"]["type"]
+                properties = feature["properties"]
+                geom = shape(feature["geometry"])
+                wkt = geom.wkt
+
+                # Insert lines
+                if geom_type == "LineString":
+                    cur.execute("""
+                        INSERT INTO osm_lines (geometry, tags)
+                        VALUES (ST_GeomFromText(%s, 4326), %s)
+                    """, (wkt, json.dumps(properties)))
+
+                # Insert polygons AND multipolygons
+                elif geom_type in ("Polygon", "MultiPolygon"):
+                    cur.execute("""
+                        INSERT INTO osm_polygons (geometry, tags)
+                        VALUES (ST_GeomFromText(%s, 4326), %s)
+                    """, (wkt, json.dumps(properties)))
+
+                else:
+                    print(f"Skipping unsupported geometry type: {geom_type}")
+
+            except Exception as e:
+                print(f"Error processing feature {idx}: {e}")
+                continue
+
+        conn.commit()
+
+    except json.JSONDecodeError:
+        print("Invalid JSON file. Ensure filtered.geojson is correctly generated.")
+    finally:
+        cur.close()
+        conn.close()
+
+if __name__ == "__main__":
+    import_geojson("israel.geojson")