feat: 导入福州历史数据

web/commons/models/fujian_survey.py

@@ -3,10 +3,12 @@ import datetime
 from dateutil.relativedelta import relativedelta
 from sqlalchemy import Column, Integer, String, Date, UniqueConstraint, Numeric
 
+from commons.models.mixin.base import BaseModelMixin
+from commons.models.model import Model
 from core.extensions import db
 
 
-class FujianSurvey(db.Model):
+class FujianSurvey(db.Model, Model, BaseModelMixin):
     __tablename__ = 'FUJIAN_SURVEY'
     id = Column('ID', Integer, primary_key=True)
     name = Column('NAME', String(128), default='', comment='名称')
@@ -20,10 +22,19 @@ class FujianSurvey(db.Model):
     region = Column('REGION', String(128), comment='地区')
 
     __table_args__ = (
-        UniqueConstraint(name, spec, date, name='Idx_key'),
+        UniqueConstraint(name, spec, date, region, name='Idx_key'),
         {'comment': '福建省交通工程材料调查表'},
     )
 
+    def find_by_key(self):
+        query = FujianSurvey.query
+        query = query.filter(FujianSurvey.name == self.name)
+        query = query.filter(FujianSurvey.spec == self.spec)
+        query = query.filter(FujianSurvey.region == self.region)
+        query = query.filter(FujianSurvey.date == self.date)
+        result = query.one_or_none()
+        return result
+
     @classmethod
     def get_query(cls, year=None, month=None, name=None, spec=None, name_in=None, region='福州'):
         query = cls.query