diff --git a/src/db/db.py b/src/db/db.py index 8a0daac7..aa78b037 100644 --- a/src/db/db.py +++ b/src/db/db.py @@ -25,8 +25,8 @@ def __init__(self): self.views = int(self.__db["views"]["count"]) self.views_from_date = self.__db["views"]["from"] self.views_to_date = self.__db["views"]["to"] - self.pull_requests = int(self.__db["pull_requests"]["count"]) - self.issues = int(self.__db["issues"]["count"]) + self.pull_requests = int(self.__db["pull_requests"]) + self.issues = int(self.__db["issues"]) def __update_db(self) -> None: try: @@ -52,9 +52,9 @@ def set_views_to_date(self, date: str) -> None: self.__update_db() def set_pull_requests(self, pull_requests_count: int) -> None: - self.__db["pull_requests"]["count"] = str(pull_requests_count) + self.__db["pull_requests"] = str(pull_requests_count) self.__update_db() def set_issues(self, issues_count: int) -> None: - self.__db["issues"]["count"] = str(issues_count) + self.__db["issues"] = str(issues_count) self.__update_db()