feat(server): add tag to imported document

papis_zotero/__init__.py

@@ -1,3 +1,4 @@
+from functools import partial
 import os
 import http.server
 from typing import Optional
@@ -24,17 +25,18 @@ def main() -> None:
               default=papis_zotero.server.ZOTERO_PORT,
               type=int)
 @click.option("--address", help="Address to bind", default="localhost")
-def serve(address: str, port: int) -> None:
+@click.option("-t", "--add-tag", help="Tag that is added to the document", default=None)
+def serve(address: str, port: int, add_tag: Optional[str]) -> None:
     """Start a ``zotero-connector`` server."""
 
     logger.warning("The 'zotero-connector' server is experimental. "
                    "Please report bugs and improvements at "
                    "https://github.com/papis/papis-zotero/issues.")
 
     server_address = (address, port)
+    request_handler = partial(papis_zotero.server.PapisRequestHandler, add_tag)
     try:
-        httpd = http.server.HTTPServer(server_address,
-                                       papis_zotero.server.PapisRequestHandler)
+        httpd = http.server.HTTPServer(server_address, request_handler)
     except OSError:
         logger.error(
             "Address '%s:%s' is already in use. This may be because you "

papis_zotero/server.py

@@ -122,6 +122,11 @@ def download_zotero_attachments(attachments: List[Dict[str, str]]) -> List[str]:
 
 
 class PapisRequestHandler(http.server.BaseHTTPRequestHandler):
+    def __init__(self, add_tag: str, request: Any,
+                 client_address: Any, server: Any) -> None:
+        self.add_tag: str = add_tag
+        super().__init__(request, client_address, server)
+
     def log_message(self, fmt: str, *args: Any) -> None:
         logger.info(fmt, *args)
 
@@ -209,6 +214,11 @@ def handle_post_add(self) -> None:
                 files = []
 
             papis_item = zotero_data_to_papis_data(item)
+            if papis_item.get("tags"):
+                papis_item["tags"].append(self.add_tag)
+            else:
+                papis_item["tags"] = [self.add_tag]
+
             logger.info("Adding paper to papis.")
             papis.commands.add.run(files, data=papis_item)