from sqlalchemy import Column, Integer, Float, String
from app.database import Base


class AtmoCompany(Base):
    """
    大气涉重企业数据模型

    @param {int} id - 污染源序号（主键，自增）
    @param {float} longitude - 企业经度坐标（精确到小数点后六位）
    @param {float} latitude - 企业纬度坐标（精确到小数点后六位）
    @param {str} company_name - 企业名称
    @param {str} company_type - 企业类型（行业分类）
    @param {str} county - 所属区县
    @param {float} particulate_emission - 大气颗粒物排放量（吨/年）
    @param {geometry} geom - 空间几何点位（点数据）
    """
    __tablename__ = 'atmo_company'

    id = Column('id', Integer, primary_key=True, autoincrement=True, comment='污染源序号')
    longitude = Column('longitude', Float, nullable=True, comment='经度坐标（精确到小数点后六位）')
    latitude = Column('latitude', Float, nullable=True, comment='纬度坐标（精确到小数点后六位）')
    company_name = Column('company_name', String(500), nullable=True, comment='企业名称')
    company_type = Column('company_type', String(500), nullable=True, comment='企业类型')
    county = Column('county', String(50), nullable=True, comment='所属区县')
    particulate_emission = Column('particulate_emission', Float, nullable=True, comment='大气颗粒物排放量（吨/年）')