diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-11-25 22:19:15 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-11-25 22:19:15 +0100 |
| commit | ee3d0d98e19f1d8177d85af1866fd0ee431fe9ea (patch) | |
| tree | 41372528e78d4328bc2a47bbbabac7e809c58894 /s2-fetch-pdf.py | |
| parent | 255b8178af1e25a71fd23703d30c0d1f74911f47 (diff) | |
moving stuff
Diffstat (limited to 's2-fetch-pdf.py')
| -rw-r--r-- | s2-fetch-pdf.py | 49 |
1 files changed, 0 insertions, 49 deletions
diff --git a/s2-fetch-pdf.py b/s2-fetch-pdf.py deleted file mode 100644 index 7d834ada..00000000 --- a/s2-fetch-pdf.py +++ /dev/null @@ -1,49 +0,0 @@ -import os -import sys -import csv -import subprocess -import time -import random -import re -import json -import click -from s2 import SemanticScholarAPI -from util import * - -s2 = SemanticScholarAPI() - -@click.command() -@click.option('--fn', '-i', default='db_paper_pdf.csv', help='Filename of CSV (id, url,)') -def fetch_pdfs(fn): - lines = read_csv(fn, keys=False) - for line in lines: - paper_id, url = line - fetch_pdf(paper_id, url) - print("{} papers processed".format(len(lines))) - -def fetch_pdf(paper_id, url): - os.makedirs(make_pdf_path(paper_id), exist_ok=True) - pdf_fn = make_pdf_fn(paper_id) - txt_fn = make_txt_fn(paper_id) - if os.path.exists(pdf_fn) or os.path.exists(txt_fn): - # return read_json(pdf_fn) - return - size = s2.fetch_file(url, pdf_fn) - if size is None: - print("{} empty?".format(paper_id)) - time.sleep(random.randint(5, 10)) - return None - print("{} {} kb".format(paper_id, int(size / 1024))) - time.sleep(random.randint(5, 10)) - return - # return paper - -def make_pdf_path(paper_id): - return './datasets/s2/pdf/{}/{}'.format(paper_id[0:2], paper_id) -def make_pdf_fn(paper_id): - return './datasets/s2/pdf/{}/{}/paper.pdf'.format(paper_id[0:2], paper_id) -def make_txt_fn(paper_id): - return './datasets/s2/pdf/{}/{}/paper.txt'.format(paper_id[0:2], paper_id) - -if __name__ == '__main__': - fetch_pdfs() |
