From 2a8587d6ae3f43a56c893c68647206357789a882 Mon Sep 17 00:00:00 2001
From: Alexandre Teles <alexandre.teles@ufba.br>
Date: Thu, 5 Jan 2023 20:01:28 -0300
Subject: [PATCH] fix(ballot): fix model references

---
 app/controllers/Ballot.py | 10 ++++++++--
 app/routers/ballot.py     |  2 +-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/app/controllers/Ballot.py b/app/controllers/Ballot.py
index 5e357d9..10584eb 100644
--- a/app/controllers/Ballot.py
+++ b/app/controllers/Ballot.py
@@ -2,6 +2,7 @@ from redis import asyncio as aioredis
 import app.utils.Logger as Logger
 from app.dependencies import load_config
 from app.utils.RedisConnector import RedisConnector
+from app.models.BallotModel import BallotModel
 
 config: dict = load_config()
 
@@ -12,7 +13,7 @@ class Ballot:
     
     BallotLogger = Logger.BallotLogger()
     
-    async def store(self, discord_hashed_id: str, ballot: str) -> bool:
+    async def store(self, discord_hashed_id: str, ballot: BallotModel) -> bool:
         """Store a ballot.
         
         Args:
@@ -26,7 +27,12 @@ class Ballot:
         stored: bool = False
         
         try:
-            await self.redis.set(name=discord_hashed_id, value=ballot, nx=True)
+            await self.redis.json().set(
+                name=discord_hashed_id,
+                path=".",
+                obj=ballot,
+                nx=True
+                )
             await self.BallotLogger.log("STORE_BALLOT", None, discord_hashed_id)
             stored = True
         except aioredis.RedisError as e:
diff --git a/app/routers/ballot.py b/app/routers/ballot.py
index 5286e12..2899f67 100644
--- a/app/routers/ballot.py
+++ b/app/routers/ballot.py
@@ -36,7 +36,7 @@ async def cast_ballot(request: Request, response: Response,
         
         stored: bool = await ballot_controller.store(
             Authorize.get_paseto_claims()['discord_hashed_id'],
-            ballot.vote
+            ballot
             )
             
         if stored: