From ec03e5e412f2a45f6fadd4e93c0f140096f7db51 Mon Sep 17 00:00:00 2001 From: simon Date: Tue, 17 Dec 2019 10:53:32 -0500 Subject: [PATCH] fix --- setup.py | 5 ++++- vanwanet_scrape/scraper.py | 3 ++- 2 files changed, 6 insertions(+), 2 deletions(-) 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.+$')