database/Plant.py at master · CapivaraProjects/database · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
from sqlalchemy import Column, String, Integer
from sqlalchemy.orm import relationship
import database.Disease
import database.Classifier
import models.Plant
from repository.base import Base


class Plant(Base.Base):
    __tablename__ = 'plants'

    id = Column(Integer, primary_key=True)
    scientificName = Column('scientific_name', String(2000))
    commonName = Column('common_name', String(2000))
    diseases = relationship('database.Disease.Disease', lazy='subquery',
                            back_populates='plant')
    classifiers = relationship('database.Classifier.Classifier', lazy='subquery',
                            back_populates='plant')

    def __init__(self,
                 id=0,
                 scientificName="",
                 commonName="",
                 diseases=[],
                 plant=models.Plant.Plant()):
        if(plant.commonName or plant.id):
            self.id = plant.id
            self.scientificName = plant.scientificName
            self.commonName = plant.commonName
            self.diseases = []
            for disease in plant.diseases:
                self.diseases.append(database.Disease.Disease(disease))
        else:
            self.id = id
            self.scientificName = scientificName
            self.commonName = commonName
            self.diseases = diseases