diff --git a/main.py b/main.py index 18882c3..7e1b53f 100644 --- a/main.py +++ b/main.py @@ -46,7 +46,7 @@ def program(): xhs = XHS(**Settings().run()) if ids := Batch().read_txt(): for i in ids: - xhs.extract(i.rstrip('\n'), download=True) + xhs.extract(i, download=True) else: while True: if url := input("请输入小红书作品链接:"): diff --git a/source/Settings.py b/source/Settings.py index d459229..cb445ad 100644 --- a/source/Settings.py +++ b/source/Settings.py @@ -35,5 +35,5 @@ class Batch: def read_txt(self) -> list: if self.file.is_file(): with self.file.open("r") as f: - return f.readlines() + return [i.rstrip('\n') for i in f.readlines()] return []