Dificento/its_day_of_the_week
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

nonetype bugfixes

Browse Source
This commit is contained in:
ilusha
2022-04-12 21:26:52 +03:00
parent b3cd6512e3
commit e029918359
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
DayOfTheWeek.py
+4 -4
View File
@@ -17,28 +17,28 @@ class Meme:
def get_random_attachment(self) -> str or None: def get_random_attachment(self) -> str or None:
attachments = self.meta.get('attachments') attachments = self.meta.get('attachments')
if len(attachments) < 1: if not attachments:
return None return None
return random.choice(self.flatten_meta(attachments)) return random.choice(self.flatten_meta(attachments))
def get_random_audio(self) -> str or None: def get_random_audio(self) -> str or None:
attachments = self.meta.get('attachments') attachments = self.meta.get('attachments')
if len(attachments) < 1 or attachments.get('audio') is None: if not attachments or attachments.get('audio') is None:
return None return None
return random.choice(attachments.get('audio')) return random.choice(attachments.get('audio'))
def get_random_video(self) -> str or None: def get_random_video(self) -> str or None:
attachments = self.meta.get('attachments') attachments = self.meta.get('attachments')
if len(attachments) < 1 or attachments.get('video') is None: if not attachments or attachments.get('video') is None:
return None return None
return random.choice(attachments.get('video')) return random.choice(attachments.get('video'))
@staticmethod @staticmethod
def flatten_meta(meta: dict[str, any]) -> list[str]: def flatten_meta(meta: dict[str, any]) -> list[str]:
return [a for k in [i for i in meta] for a in k] return [a for k in [i for i in meta.values()] for a in k]
class Day: class Day: