update nyaasi API
This commit is contained in:
@@ -2,6 +2,12 @@
|
|||||||
from komsuite_nyaapy import Nyaa
|
from komsuite_nyaapy import Nyaa
|
||||||
from natsort import natsorted
|
from natsort import natsorted
|
||||||
import re
|
import re
|
||||||
|
from komconfig import KomConfig
|
||||||
|
import loguru
|
||||||
|
|
||||||
|
log = loguru.logger
|
||||||
|
log.add("logs/nyaasi.log", rotation="1 week")
|
||||||
|
config = KomConfig()
|
||||||
|
|
||||||
|
|
||||||
class NyaaFeed(Nyaa):
|
class NyaaFeed(Nyaa):
|
||||||
@@ -21,18 +27,25 @@ class NyaaFeed(Nyaa):
|
|||||||
return natsorted(volumes)
|
return natsorted(volumes)
|
||||||
|
|
||||||
def search(self, title: str):
|
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)
|
dataset = super().search(title, 3, 1)
|
||||||
datalist = []
|
datalist = []
|
||||||
if dataset is None:
|
if dataset is None:
|
||||||
return datalist
|
return datalist
|
||||||
for entry in dataset:
|
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 = {}
|
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
|
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"])
|
data["volumes"] = re.findall(regex, data["title"])
|
||||||
try:
|
try:
|
||||||
match = re.match(
|
match = re.match(
|
||||||
@@ -52,9 +65,9 @@ class NyaaFeed(Nyaa):
|
|||||||
data["volumes"] = (
|
data["volumes"] = (
|
||||||
self.list_volumes(data["volumes"]) if data["volumes"] else [0]
|
self.list_volumes(data["volumes"]) if data["volumes"] else [0]
|
||||||
)
|
)
|
||||||
data["link"] = entry.link
|
data["link"] = entry.download_url
|
||||||
data["seeders"] = entry.nyaa_seeders
|
data["seeders"] = entry.seeders
|
||||||
data["filesize"] = entry.nyaa_size
|
data["filesize"] = entry.size
|
||||||
# print(data)
|
# print(data)
|
||||||
datalist.append(data)
|
datalist.append(data)
|
||||||
|
|
||||||
@@ -63,3 +76,4 @@ class NyaaFeed(Nyaa):
|
|||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
print("This is a module, not a script")
|
print("This is a module, not a script")
|
||||||
|
print(NyaaFeed().search("One Piece"))
|
||||||
|
|||||||
Reference in New Issue
Block a user