59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
from NyaaPy.sukebei import SukebeiNyaa
|
|
from datetime import datetime
|
|
import json
|
|
import os
|
|
|
|
# Creating a folder for test_files
|
|
# ! not included in github project.
|
|
if not os.path.isdir("test_files"):
|
|
os.makedirs("test_files")
|
|
|
|
nyaa = SukebeiNyaa()
|
|
|
|
# Get fresh torrents
|
|
dt_latest_torrents_begin = datetime.now()
|
|
latest_torrents = nyaa.last_uploads(100)
|
|
dt_latest_torrents_end = datetime.now()
|
|
with open("test_files/sukebei_latest_torrent_test.json", 'w') as f:
|
|
json.dump(latest_torrents, f)
|
|
|
|
# Search some nasty stuff
|
|
dt_search_begin = datetime.now()
|
|
test_search = nyaa.search("G Senjou no maou")
|
|
dt_search_end = datetime.now()
|
|
with open("test_files/sukebei_search_test.json", 'w') as f:
|
|
json.dump(test_search, f)
|
|
|
|
# Get first torrent from found torrents
|
|
dt_single_torrent_begin = datetime.now()
|
|
single_torrent = nyaa.get(test_search[0]["id"])
|
|
dt_single_torrent_end = datetime.now()
|
|
with open("test_files/sukebei_single_torrent_test.json", 'w') as f:
|
|
json.dump(single_torrent, f)
|
|
|
|
dt_user_begin = datetime.now()
|
|
user_torrents = nyaa.get_user("RUNBKK")
|
|
dt_user_end = datetime.now()
|
|
with open("test_files/sukebei_single_user_test.json", 'w') as f:
|
|
json.dump(user_torrents, f)
|
|
|
|
print(
|
|
"Latest torrents time:",
|
|
(dt_latest_torrents_end - dt_latest_torrents_begin).microseconds / 1000,
|
|
"msec")
|
|
print(
|
|
"Test search time:",
|
|
(dt_search_end - dt_search_begin).microseconds / 1000,
|
|
"msec"
|
|
)
|
|
print(
|
|
"Single torrent time:",
|
|
(dt_single_torrent_end - dt_single_torrent_begin).microseconds / 1000,
|
|
"msec"
|
|
)
|
|
print(
|
|
"Single user time:",
|
|
(dt_user_end - dt_user_begin).microseconds / 1000,
|
|
"msec"
|
|
)
|