Make into package

kermitt2 · Mar 29, 2021 · 61fa90f · 61fa90f
1 parent 5571763
commit 61fa90f
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 2 deletions.
diff --git a/__init__.py → grobid_client/__init__.py b/__init__.py → grobid_client/__init__.py
diff --git a/client.py → grobid_client/client.py b/client.py → grobid_client/client.py
diff --git a/grobid_client.py → grobid_client/grobid_client.py b/grobid_client.py → grobid_client/grobid_client.py
@@ -22,7 +22,7 @@
 import requests
 import pathlib
 
-from client import ApiClient
+from .client import ApiClient
 
 
 class GrobidClient(ApiClient):
@@ -258,7 +258,7 @@ def process_pdf(
         return (pdf_file, status, res.text)
 
 
-if __name__ == "__main__":
+def main():
     valid_services = [
         "processFulltextDocument",
         "processHeaderDocument",
@@ -385,3 +385,6 @@ def process_pdf(
 
     runtime = round(time.time() - start_time, 3)
     print("runtime: %s seconds " % (runtime))
+
+if __name__ == "__main__":
+    main()
diff --git a/test.py → grobid_client/test.py b/test.py → grobid_client/test.py
diff --git a/setup.py b/setup.py
@@ -7,5 +7,8 @@
       description='grobid_client_python',
       author='kermitt2',
       packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),
+      entry_points={
+          'console_scripts': ['grobid_client=grobid_client.grobid_client:main']
+      },
       license='LICENSE',
     )