diff --git a/curse-stats.py b/curse-stats.py index d5dd59e..2bd4248 100755 --- a/curse-stats.py +++ b/curse-stats.py @@ -12,10 +12,10 @@ from datetime import datetime, timedelta, timezone class Mod: def __init__(self, data): - self.id = data["id"] - self.name = data["name"] - self.downloads = data["downloadCount"] - self.date_created = data["dateCreated"] + self.id = int(data["id"]) + self.name = str(data["name"]) + self.downloads = int(data["downloadCount"]) + self.date_created = str(data["dateCreated"]) year_created = "" try: year_created = datetime.strptime( @@ -26,7 +26,7 @@ class Mod: data["dateCreated"], "%Y-%m-%dT%H:%M:%S%z" ).strftime("%Y") self.year_created = year_created - self.date_modified = data["dateModified"] + self.date_modified = str(data["dateModified"]) try: self.date_released = datetime.strptime( data["dateReleased"], "%Y-%m-%dT%H:%M:%S.%f%z" @@ -100,14 +100,16 @@ if __name__ == "__main__": # Merge all filters into 1 array regardless of the arg format filters = [] - for filter_list in args.filter: - for filter in filter_list: - filters.append(filter) + if args.filter is not None: + for filter_list in args.filter: + for filter in filter_list: + filters.append(filter) sorts = [] - for sort_list in args.sort: - for sort_logic in sort_list: - sorts.append(sort_logic) + if args.sort is not None: + for sort_list in args.sort: + for sort_logic in sort_list: + sorts.append(sort_logic) def sort_mod(mod: Mod, sorts: list[SortType]): sorting = [] @@ -147,7 +149,7 @@ if __name__ == "__main__": except ValueError as err: print(err) break - # pagination = data['pagination'] + data = data["data"] for entry in data: