feat: support priority

Author: Zxilly

poetry.lock

@@ -1,9 +1,9 @@
 [tool.poetry]
 name = "znotify"
-version = "0.2.0"
+version = "0.3.0"
 description = "sdk for notify"
 authors = ["Zxilly <[email protected]>"]
-license = "GPL-3.0"
+license = "Apache-2.0"
 readme = "README.md"
 repository = "https://github.com/ZNotify/py-sdk"
 classifiers = [
@@ -23,6 +23,9 @@ requests = "^2.27.1"
 [tool.poetry.urls]
 "Bug Tracker" = "https://github.com/ZNotify/py-sdk/issues"
 
+[tool.poetry.group.dev.dependencies]
+poethepoet = "^0.16.5"
+
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"

pyproject.toml

@@ -1,4 +1,6 @@
 import unittest
+import warnings
+
 from znotify import Client
 
 
@@ -53,3 +55,4 @@ def test_client_send_3(self):
 
 if __name__ == "__main__":
     unittest.main()
+

tests/test_client.py

@@ -1,4 +1,5 @@
 import unittest
+
 from znotify.send import send
 
 

tests/test_send.py

@@ -1,2 +1,4 @@
 from .client import *
 from .send import *
+
+from .version import __title__, __description__, __url__, __version__, __author__, __license__

znotify/__init__.py

@@ -1,41 +1,46 @@
 import requests
-from requests.exceptions import ConnectionError
 
-from znotify.static import ENDPOINT
+from znotify.static import ENDPOINT, Priority
+from znotify.version import __version__
 
 
 class Client:
-    def __init__(self, user_id, endpoint):
+    def __init__(self, user_id: str, endpoint: str):
         self.endpoint = endpoint if endpoint else ENDPOINT
         self.user_id = user_id
+        self.session = requests.Session()
+        self.session.headers.update({
+            "User-Agent": f"znotify-py-sdk/{__version__}",
+        })
 
     @staticmethod
-    def create(user_id, endpoint=None):
+    def create(user_id: str, endpoint: str = None) -> "Client":
         client = Client(user_id, endpoint)
         client.check()
         return client
 
     def check(self):
-        resp = requests.get(f"{self.endpoint}/check", params={"user_id": self.user_id})
+        resp = self.session.get(f"{self.endpoint}/check", params={"user_id": self.user_id})
         if not resp.json()["body"]:
             raise Exception("User ID not valid")
 
-    def send(self, content, title=None, long=None):
+    def send(self, content: str, title: str = None, long: str = None, priority: Priority = None):
         if content is None or content == "":
             raise Exception("Content is required")
 
         if title is None:
             title = "Notification"
         if long is None:
             long = ""
+        if priority is None:
+            priority = Priority.NORMAL
 
         data = {
             "title": title,
             "content": content,
-            "long": long
+            "long": long,
+            "priority": priority.value,
         }
-        try:
-            resp = requests.post(f"{self.endpoint}/{self.user_id}/send", data=data)
-        except ConnectionError as e:
-            raise Exception("Connection error") from e
+
+        resp = self.session.post(f"{self.endpoint}/{self.user_id}/send", data=data)
         return resp.json()["body"]

znotify/client.py

@@ -1,3 +1,10 @@
 import sys
+from enum import Enum
 
 ENDPOINT = "https://push.learningman.top" if 'unittest' not in sys.modules else "http://localhost:14444"
+
+
+class Priority(Enum):
+    HIGH = "high"
+    NORMAL = "normal"
+    LOW = "low"

znotify/static.py

@@ -0,0 +1,6 @@
+__title__ = "znotify"
+__description__ = "Python SDK for ZNotify."
+__url__ = "https://github.com/ZNotify/py-sdk"
+__version__ = "0.3.0"
+__author__ = "Zxilly"
+__license__ = "Apache-2.0"