diff --git a/setup.py b/setup.py index e581832..a32ca50 100644 --- a/setup.py +++ b/setup.py @@ -9,5 +9,8 @@ setup( packages=["vanwanet_scrape"], install_requires=[ "requests", "bs4", "hexlib @ git+git://github.com/simon987/hexlib.git", - ] + ], + package_data={ + "vanwanet_scrape": ["*.js"], + } ) diff --git a/vanwanet_scrape/scraper.py b/vanwanet_scrape/scraper.py index 4f19e07..6c506ff 100644 --- a/vanwanet_scrape/scraper.py +++ b/vanwanet_scrape/scraper.py @@ -1,3 +1,4 @@ +import os import re import subprocess from http.cookiejar import CookieJar @@ -6,7 +7,7 @@ import requests from bs4 import BeautifulSoup from hexlib.web import cookie_from_string -with open("aes.js", "r") as f: +with open(os.path.join(os.path.dirname(__file__), "aes.js"), "r") as f: AES = f.read() SUB_PATTRN = re.compile(r'document\.cookie="(.+)";location.+$')