From 46861eac863b6e211d41fafad9a174704c313aad Mon Sep 17 00:00:00 2001 From: Ken Kellner Date: Thu, 16 Jan 2020 09:05:15 -0500 Subject: Add Windows compatibility, thanks to @kebasaa --- article_epub/publisher.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/article_epub/publisher.py b/article_epub/publisher.py index 8f67f78..c240400 100644 --- a/article_epub/publisher.py +++ b/article_epub/publisher.py @@ -8,6 +8,7 @@ from time import sleep import subprocess import requests import json +import tempfile _publishers = list() _publisher_domains = dict() @@ -31,10 +32,17 @@ class Publisher(object): self.get_final_url() os.environ['MOZ_HEADLESS'] = '1' print('Starting headless browser...',end='',flush=True) - binary = FirefoxBinary('/usr/bin/firefox') + + if(os.name == "posix"): + binary = FirefoxBinary('/usr/bin/firefox') + elif(os.name == "nt"): + binary = FirefoxBinary('C:/Program Files/Mozilla Firefox/firefox.exe') + else: + sys.exit("Error: Unknown OS") + try: driver = webdriver.Firefox(firefox_binary=binary, - log_path='/tmp/gecko_log') + log_path=os.path.join(tempfile.gettempdir(), 'gecko_log')) print('done') except: sys.exit('Failed to load Firefox; is it installed?') @@ -151,7 +159,7 @@ class Publisher(object): else: self.output = output - output_raw = '/tmp/raw.epub' + output_raw = os.path.join(tempfile.gettempdir(), 'raw.epub') combined = '' combined += str(self.get_citation(link=True)) -- cgit v1.2.3