update nyaasi API
This commit is contained in:
@@ -2,6 +2,12 @@
|
||||
from komsuite_nyaapy import Nyaa
|
||||
from natsort import natsorted
|
||||
import re
|
||||
from komconfig import KomConfig
|
||||
import loguru
|
||||
|
||||
log = loguru.logger
|
||||
log.add("logs/nyaasi.log", rotation="1 week")
|
||||
config = KomConfig()
|
||||
|
||||
|
||||
class NyaaFeed(Nyaa):
|
||||
@@ -21,18 +27,25 @@ class NyaaFeed(Nyaa):
|
||||
return natsorted(volumes)
|
||||
|
||||
def search(self, title: str):
|
||||
regex = r"\b(v\d{2,3}-\d{2,3}|v\d{2,3}-v\d{2,3}|\d{2}|\d{2}-\d{2}|v\d{2,3})\b"
|
||||
dataset = super().search(title, 3, 1)
|
||||
datalist = []
|
||||
if dataset is None:
|
||||
return datalist
|
||||
for entry in dataset:
|
||||
if config.komgrabber.get_chapters is False:
|
||||
if "chapter" in entry.name.lower():
|
||||
log.info(
|
||||
"Skipping {}, Reason: Chapters disabled".format(entry.name)
|
||||
)
|
||||
continue
|
||||
data = {}
|
||||
# print(entry)
|
||||
# if not "digital" in entry.title.lower():
|
||||
# # print("Skipping {}".format(entry.title))
|
||||
# # log.log("Skipping {}".format(entry.title))
|
||||
# continue
|
||||
data["title"] = entry.name
|
||||
if any(
|
||||
x in data["title"].lower() for x in config.komgrabber.skip_parameters
|
||||
):
|
||||
log.info("Skipping {}".format(data["title"]))
|
||||
continue
|
||||
data["volumes"] = re.findall(regex, data["title"])
|
||||
try:
|
||||
match = re.match(
|
||||
@@ -52,9 +65,9 @@ class NyaaFeed(Nyaa):
|
||||
data["volumes"] = (
|
||||
self.list_volumes(data["volumes"]) if data["volumes"] else [0]
|
||||
)
|
||||
data["link"] = entry.link
|
||||
data["seeders"] = entry.nyaa_seeders
|
||||
data["filesize"] = entry.nyaa_size
|
||||
data["link"] = entry.download_url
|
||||
data["seeders"] = entry.seeders
|
||||
data["filesize"] = entry.size
|
||||
# print(data)
|
||||
datalist.append(data)
|
||||
|
||||
@@ -63,3 +76,4 @@ class NyaaFeed(Nyaa):
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("This is a module, not a script")
|
||||
print(NyaaFeed().search("One Piece"))
|
||||
|
||||
Reference in New Issue
Block a user