1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
#!/usr/bin/env python3
# programmes expire, so i have to update the times in the tests every time i run them
# but thats a massive ballache, so i end up just not running them, which leads to cockups
# so, this script has the tests automatically use the latest episode as you run it, by setting dynamically generated time values
# everything else is always the same so it should be fine lol
import datetime
import os
import sys
import unittest
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
sys.path.insert(0, "/home/g/Downloads/yt-dlp/") # TODO: un-hardcode. has to be the source/git repo because pip doesnt carry the tests
from yt_dlp_plugins.extractor import radiko_time as rtime
from yt_dlp_plugins.extractor.radiko import RadikoTimeFreeIE
MON, TUE, WED, THU, FRI, SAT, SUN = range(7)
weekdays = {0: "MON", 1: "TUE", 2: "WED", 3: "THU", 4: "FRI", 5: "SAT", 6: "SUN"}
now = rtime.RadikoTime.now(tz = rtime.JST)
UTC = datetime.timezone.utc
def get_latest_airtimes(now, weekday, hour, minute, duration):
days_after_weekday = (7 - (now.weekday() - weekday)) % 7
latest_airdate = (now + datetime.timedelta(days=days_after_weekday)).replace(hour=hour, minute=minute, second=0, microsecond=0)
if (latest_airdate + duration) > now:
latest_airdate -= datetime.timedelta(days=7)
return latest_airdate, latest_airdate + duration
def get_test_timefields(airtime, release_time):
return {
"timestamp": airtime.timestamp(),
"release_timestamp": release_time.timestamp(),
"upload_date": airtime.astimezone(UTC).strftime("%Y%m%d"),
"release_date": release_time.astimezone(UTC).strftime("%Y%m%d"),
"duration": (release_time - airtime).total_seconds(),
}
RadikoTimeFreeIE._TESTS = []
# TOKYO MOON - interfm - EVERY FRI 2300
airtime, release_time = get_latest_airtimes(now, FRI, 23, 0, datetime.timedelta(hours=1))
RadikoTimeFreeIE._TESTS.append({
"url": f"https://radiko.jp/#!/ts/INT/{airtime.timestring()}",
"info_dict": {
"ext": "m4a",
"id": f"INT-{airtime.timestring()}",
**get_test_timefields(airtime, release_time),
'title': 'TOKYO MOON',
'description': 'md5:4185349a530cfc4d0580e6996a511273',
'uploader': 'interfm',
'uploader_id': 'INT',
'uploader_url': 'https://www.interfm.co.jp/',
'channel': 'interfm',
'channel_id': 'INT',
'channel_url': 'https://www.interfm.co.jp/',
'thumbnail': 'https://program-static.cf.radiko.jp/ehwtw6mcvy.jpg',
'chapters': list,
'tags': ['松浦俊夫', 'ジャズの魅力を楽しめる'],
'cast': ['松浦\u3000俊夫'],
'series': 'Tokyo Moon',
'live_status': 'was_live',
}
})
IEs = [RadikoTimeFreeIE]
import test.helper as th
# override to only get testcases from our IEs
def _new_gettestcases(include_onlymatching=False):
import yt_dlp.plugins as plugins
plugins.load_all_plugins()
for ie in IEs:
yield from ie.get_testcases(include_onlymatching)
def _new_getwebpagetestcases():
import yt_dlp.plugins as plugins
plugins.load_all_plugins()
for ie in IEs:
for tc in ie.get_webpage_testcases():
tc.setdefault('add_ie', []).append('Generic')
yield tc
th.gettestcases = _new_gettestcases
th.getwebpagetestcases = _new_getwebpagetestcases
import test.test_download as td
class TestDownload(td.TestDownload):
pass
if __name__ == "__main__":
unittest.main()
|
