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


class AgriculturalData(Base):
    """
    农业投入品与镉输入通量数据模型

    @param {int} id - 主键ID（自增）
    @param {str} county_name - 县市名称
    @param {float} crop_sowing_area - 农作物播种面积（公顷）
    @param {float} nitrogen_usage - 氮肥使用量（吨/公顷/年）
    @param {float} phosphorus_usage - 磷肥使用量（吨/公顷/年）
    @param {float} potassium_usage - 钾肥使用量（吨/公顷/年）
    @param {float} compound_usage - 复合肥使用量（吨/公顷/年）
    @param {float} organic_usage - 有机肥使用量（吨/公顷/年）
    @param {float} pesticide_usage - 农药使用量（吨/公顷/年）
    @param {float} farmyard_usage - 农家肥使用量（吨/公顷/年）
    @param {float} plastic_film_usage - 农膜存留量（吨/公顷/年）
    @param {float} nitrogen_cd_flux - 氮肥镉输入通量（g/ha/a）
    @param {float} phosphorus_cd_flux - 磷肥镉输入通量（g/ha/a）
    @param {float} potassium_cd_flux - 钾肥镉输入通量（g/ha/a）
    @param {float} compound_cd_flux - 复合肥镉输入通量（g/ha/a）
    @param {float} organic_cd_flux - 有机肥镉输入通量（g/ha/a）
    @param {float} pesticide_cd_flux - 农药镉输入通量（g/ha/a）
    @param {float} farmyard_cd_flux - 农家肥镉输入通量（g/ha/a）
    @param {float} plastic_film_cd_flux - 农膜镉输入通量（g/ha/a）
    @param {float} total_cd_flux - 总镉输入通量（g/ha/a）
    @param {int} data_year - 数据年份
    """
    __tablename__ = 'agricultural_data'

    id = Column('ID', Integer, primary_key=True, autoincrement=True)
    county_name = Column('county_name', String(50), nullable=True, comment='县市名称')
    crop_sowing_area = Column('crop_sowing_area', Float, nullable=True, comment='农作物播种面积（公顷）')

    # 农用投入品使用量
    nitrogen_usage = Column('nitrogen_usage', Float, nullable=True, comment='氮肥使用量（吨/公顷/年）')
    phosphorus_usage = Column('phosphorus_usage', Float, nullable=True, comment='磷肥使用量（吨/公顷/年）')
    potassium_usage = Column('potassium_usage', Float, nullable=True, comment='钾肥使用量（吨/公顷/年）')
    compound_usage = Column('compound_usage', Float, nullable=True, comment='复合肥使用量（吨/公顷/年）')
    organic_usage = Column('organic_usage', Float, nullable=True, comment='有机肥使用量（吨/公顷/年）')
    pesticide_usage = Column('pesticide_usage', Float, nullable=True, comment='农药使用量（吨/公顷/年）')
    farmyard_usage = Column('farmyard_usage', Float, nullable=True, comment='农家肥使用量（吨/公顷/年）')
    plastic_film_usage = Column('plastic_film_usage', Float, nullable=True, comment='农膜存留量（吨/公顷/年）')

    # 镉输入通量
    nitrogen_cd_flux = Column('nitrogen_cd_flux', Float, nullable=True, comment='氮肥镉输入通量（g/ha/a）')
    phosphorus_cd_flux = Column('phosphorus_cd_flux', Float, nullable=True, comment='磷肥镉输入通量（g/ha/a）')
    potassium_cd_flux = Column('potassium_cd_flux', Float, nullable=True, comment='钾肥镉输入通量（g/ha/a）')
    compound_cd_flux = Column('compound_cd_flux', Float, nullable=True, comment='复合肥镉输入通量（g/ha/a）')
    organic_cd_flux = Column('organic_cd_flux', Float, nullable=True, comment='有机肥镉输入通量（g/ha/a）')
    pesticide_cd_flux = Column('pesticide_cd_flux', Float, nullable=True, comment='农药镉输入通量（g/ha/a）')
    farmyard_cd_flux = Column('farmyard_cd_flux', Float, nullable=True, comment='农家肥镉输入通量（g/ha/a）')
    plastic_film_cd_flux = Column('plastic_film_cd_flux', Float, nullable=True, comment='农膜镉输入通量（g/ha/a）')

    # 汇总指标
    total_cd_flux = Column('total_cd_flux', Float, nullable=True, comment='总镉输入通量（g/ha/a）')
    data_year = Column('data_year', Integer, nullable=True, comment='数据年份')