[utils] Fix random_birthday to generate existing dates only
authorAlexander Seiler <seileralex@gmail.com>
Sat, 1 Dec 2018 17:05:15 +0000 (18:05 +0100)
committerSergey M <dstftw@gmail.com>
Sat, 1 Dec 2018 17:05:15 +0000 (00:05 +0700)
youtube_dl/utils.py

index e84d35d4dee2077faf89bbecb0083ca01e6273c4..0b1c7cd6c7a3b3229844fa4e53299bdf1139353a 100644 (file)
@@ -3948,8 +3948,12 @@ def write_xattr(path, key, value):
 
 
 def random_birthday(year_field, month_field, day_field):
+    start_date = datetime.date(1950, 1, 1)
+    end_date = datetime.date(1995, 12, 31)
+    offset = random.randint(0, (end_date - start_date).days)
+    random_date = start_date + datetime.timedelta(offset)
     return {
-        year_field: str(random.randint(1950, 1995)),
-        month_field: str(random.randint(1, 12)),
-        day_field: str(random.randint(1, 31)),
+        year_field: str(random_date.year),
+        month_field: str(random_date.month),
+        day_field: str(random_date.day),
     }