def insert_vote(cls, candidate_id): session = DB.get_session() new_vote = Vote(reference=candidate_id, up_date=datetime.now()) session.add(new_vote) session.flush() session.refresh(new_vote) id_to_return = new_vote.id session.commit() return id_to_return
def create(cls, data): if (len( cls.find_candidate(data["first_name"] + " " + data["last_name"])) > 0): return session = DB.get_session() session.add_all([ Candidate(first_name=data["first_name"], last_name=data["last_name"], up_date=datetime.now()) ]) session.commit()
def get_candidate_votes(cls, candidate_id): print(candidate_id) return DB.get_session().query(cls).filter( Vote.reference == candidate_id).all()
def get_all(cls): return DB.get_session().query(cls).all()
def find_candidate(cls, candidate_name): first_name = candidate_name.split(" ")[0] last_name = candidate_name.split(" ")[1] return DB.get_session().query(cls).filter(Candidate.first_name == first_name, Candidate.last_name == last_name)\ .all()