Git-Mirrors/censorbib
1
0
Fork 0
mirror of https://github.com/NullHypothesis/censorbib.git synced 2025-07-25 15:35:55 -04:00
Code Issues Projects Releases Packages Wiki Activity

Port script to Python 3.

Browse source
This commit is contained in:
Philipp Winter 2022-01-21 11:30:59 -08:00
parent 8567365f53
commit de23fa79b7
1 changed files with 12 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

23
fetch_pdfs.py
View file

@ -1,4 +1,4 @@
#!/usr/bin/env python #!/usr/bin/env python3
# #
# Copyright 2015 Philipp Winter <phw@nymity.ch> # Copyright 2015 Philipp Winter <phw@nymity.ch>
# #
@ -21,7 +21,7 @@ Fetch pdf and ps files in BibTeX file.
import os import os
import sys import sys
import errno import errno
import urllib2 import urllib.request
import pybtex.database.input.bibtex as bibtex import pybtex.database.input.bibtex as bibtex
@ -31,15 +31,15 @@ def download_pdf(url, file_name):
Download file and write it to given file name. Download file and write it to given file name.
""" """
print "Now fetching %s" % url print("Now fetching %s" % url)
try: try:
fetched_file = urllib2.urlopen(url) fetched_file = urllib.request.urlopen(url)
except Exception as err: except Exception as err:
print >> sys.stderr, err print(err, file=sys.stderr)
return return
with open(file_name, "w") as fd: with open(file_name, "wb") as fd:
fd.write(fetched_file.read()) fd.write(fetched_file.read())
@ -78,14 +78,14 @@ def main(file_name, output_dir):
ext = ext[1:] ext = ext[1:]
if ext not in ["pdf", "ps"]: if ext not in ["pdf", "ps"]:
print >> sys.stderr, ("Skipping %s because it's not a pdf " print("Skipping %s because it's not a pdf or ps file." % url,
"or ps file." % url) file=sys.stderr)
continue continue
file_name = os.path.join(output_dir, ext, bibkey + ".%s" % ext) file_name = os.path.join(output_dir, ext, bibkey + ".%s" % ext)
if os.path.exists(file_name): if os.path.exists(file_name):
print >> sys.stderr, ("Skipping %s because we already " print("Skipping %s because we already have it." % file_name,
"have it." % file_name) file=sys.stderr)
continue continue
download_pdf(url, file_name) download_pdf(url, file_name)
@ -96,7 +96,8 @@ def main(file_name, output_dir):
if __name__ == "__main__": if __name__ == "__main__":
if len(sys.argv) != 3: if len(sys.argv) != 3:
print >> sys.stderr, "\nUsage: %s FILE_NAME OUTPUT_DIR\n" % sys.argv[0] print("\nUsage: %s FILE_NAME OUTPUT_DIR\n" % sys.argv[0],
file=sys.stderr)
sys.exit(1) sys.exit(1)
sys.exit(main(sys.argv[1], sys.argv[2])) sys.exit(main(sys.argv[1], sys.argv[2]))