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()