Add get_osm method to Line class

2021-07-16 11:14:49 +02:00 · 2021-07-16 11:14:49 +02:00 · a56387c4d9
parent 66904c061a
commit a56387c4d9
1 changed files with 5 additions and 0 deletions
--- a/matcher/model.py
+++ b/matcher/model.py
@ -288,6 +288,11 @@ class Line(MapMixin, Base):
    def type(self):
        return "way" if self.src_id > 0 else "relation"

+    @classmethod
+    def get_osm(cls, osm_type, osm_id):
+        src_id = osm_id * {'way': 1, 'relation': -1}[osm_type]
+        return cls.query.get(src_id)
+

 class Polygon(MapMixin, Base):
    way_area = Column(Float)