| import asyncio |
| import os |
|
|
| from api_client import get_phpsessid |
| from pixiv_api import headers, sanitize_filename, search |
|
|
| ROOT_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..")) |
| PAGES = 300 // 60 |
|
|
| input_url = input("Enter the URL: ") |
|
|
| def main() -> int: |
| phpsessid = get_phpsessid() |
| pixiv_cookies = {"PHPSESSID": phpsessid} |
| try: |
| post_ids, filename = asyncio.run(search( |
| input_url, |
| PAGES, |
| ai_only=False, |
| real_only=True, |
| cookies=pixiv_cookies, |
| headers=headers, |
| )) |
| except Exception as exc: |
| print(f"Error: {exc}") |
| return 1 |
| filename = sanitize_filename(filename) |
| with open(os.path.join(ROOT_DIR, f"{filename}.txt"), "w", encoding="utf-8") as f: |
| f.write("\n".join(map(str, post_ids))) |
| return 0 |
|
|
| if __name__ == "__main__": |
| raise SystemExit(main()) |
|
|
