diff options
| -rwxr-xr-x | arxiv2remarkable.py | 24 | ||||
| -rw-r--r-- | test.py | 24 |
2 files changed, 24 insertions, 24 deletions
diff --git a/arxiv2remarkable.py b/arxiv2remarkable.py index 8be885b..8f605ab 100755 --- a/arxiv2remarkable.py +++ b/arxiv2remarkable.py @@ -333,7 +333,7 @@ class Provider(metaclass=abc.ABCMeta): return target_path -class ArxivProvider(Provider): +class Arxiv(Provider): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) @@ -379,7 +379,7 @@ class ArxivProvider(Provider): return dict(title=title, date=date, authors=authors) -class PMCProvider(Provider): +class Pubmed(Provider): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) @@ -434,7 +434,7 @@ class PMCProvider(Provider): return dict(title=title, date=date, authors=authors) -class ACMProvider(Provider): +class ACM(Provider): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) @@ -502,7 +502,7 @@ class ACMProvider(Provider): return dict(title=title, date=date, authors=authors) -class OpenReviewProvider(Provider): +class OpenReview(Provider): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) @@ -548,7 +548,7 @@ class OpenReviewProvider(Provider): return dict(title=title, date=date, authors=authors) -class LocalFileProvider(Provider): +class LocalFile(Provider): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) @@ -568,7 +568,7 @@ class LocalFileProvider(Provider): return os.path.basename(info["filename"]) -class PdfUrlProvider(Provider): +class PdfUrl(Provider): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) @@ -805,12 +805,12 @@ def main(): args = parse_args() providers = [ - ArxivProvider, - PMCProvider, - ACMProvider, - OpenReviewProvider, - LocalFileProvider, - PdfUrlProvider, + Arxiv, + Pubmed, + ACM, + OpenReview, + LocalFile, + PdfUrl, ] provider = next((p for p in providers if p.validate(args.input)), None) @@ -12,12 +12,12 @@ import shutil import os from arxiv2remarkable import ( - ArxivProvider, - PMCProvider, - ACMProvider, - OpenReviewProvider, - LocalFileProvider, - PdfUrlProvider, + Arxiv, + Pubmed, + ACM, + OpenReview, + LocalFile, + PdfUrl, ) @@ -46,7 +46,7 @@ class Tests(unittest.TestCase): shutil.rmtree(self.test_dir) def test_arxiv(self): - prov = ArxivProvider(upload=False) + prov = Arxiv(upload=False) url = "https://arxiv.org/abs/1811.11242v1" exp_filename = "Burg_Nazabal_Sutton_-_Wrangling_Messy_CSV_Files_by_Detecting_Row_and_Type_Patterns_2018.pdf" filename = prov.run(url) @@ -55,7 +55,7 @@ class Tests(unittest.TestCase): self.assertTrue(1054082 < fsize <= 1056082) def test_pmc(self): - prov = PMCProvider(upload=False) + prov = Pubmed(upload=False) url = "https://www.ncbi.nlm.nih.gov/pmc/articles/PMC3474301/" exp_filename = ( "Hoogenboom_Manske_-_How_to_Write_a_Scientific_Article_2012.pdf" @@ -66,7 +66,7 @@ class Tests(unittest.TestCase): self.assertTrue(376640 < fsize <= 378640) def test_acm(self): - prov = ACMProvider(upload=False) + prov = ACM(upload=False) url = "https://dl.acm.org/citation.cfm?id=3025626" exp_filename = "Kery_Horvath_Myers_-_Variolite_Supporting_Exploratory_Programming_by_Data_Scientists_2017.pdf" filename = prov.run(url) @@ -75,7 +75,7 @@ class Tests(unittest.TestCase): self.assertTrue(2349734 < fsize <= 2351734) def test_openreview(self): - prov = OpenReviewProvider(upload=False) + prov = OpenReview(upload=False) url = "https://openreview.net/forum?id=S1x4ghC9tQ" exp_filename = "Gregor_et_al_-_Temporal_Difference_Variational_Auto-Encoder_2018.pdf" filename = prov.run(url) @@ -89,14 +89,14 @@ class Tests(unittest.TestCase): fp.write( "%PDF-1.1\n%¥±ë\n\n1 0 obj\n << /Type /Catalog\n /Pages 2 0 R\n >>\nendobj\n\n2 0 obj\n << /Type /Pages\n /Kids [3 0 R]\n /Count 1\n /MediaBox [0 0 300 144]\n >>\nendobj\n\n3 0 obj\n << /Type /Page\n /Parent 2 0 R\n /Resources\n << /Font\n << /F1\n << /Type /Font\n /Subtype /Type1\n /BaseFont /Times-Roman\n >>\n >>\n >>\n /Contents 4 0 R\n >>\nendobj\n\n4 0 obj\n << /Length 55 >>\nstream\n BT\n /F1 18 Tf\n 0 0 Td\n (Hello World) Tj\n ET\nendstream\nendobj\n\nxref\n0 5\n0000000000 65535 f \n0000000018 00000 n \n0000000077 00000 n \n0000000178 00000 n \n0000000457 00000 n \ntrailer\n << /Root 1 0 R\n /Size 5\n >>\nstartxref\n565\n%%EOF" ) - prov = LocalFileProvider(upload=False) + prov = LocalFile(upload=False) filename = prov.run(local_filename) self.assertEqual("test_.pdf", os.path.basename(filename)) fsize = os.path.getsize(filename) self.assertTrue(5843 < fsize <= 7843) def test_pdfurl(self): - prov = PdfUrlProvider(upload=False) + prov = PdfUrl(upload=False) url = "http://www.jmlr.org/papers/volume17/14-526/14-526.pdf" filename = prov.run(url, filename="test.pdf") self.assertEqual("test.pdf", os.path.basename(filename)) |