Merge pull request #14 from uJhin/1.2.2

Upbit OPEN API v1.2.2

Author: uJhin

requirements.txt

@@ -44,5 +44,5 @@ typing-extensions==3.7.4.3
 urllib3==1.26.4
 webcolors==1.11.1
 webencodings==0.5.1
-websockets==8.1
+websockets==10.1
 wrapt==1.12.1

upbit/pkginfo.py

@@ -4,7 +4,7 @@
 Please read the official Upbit Client document.
 Documents: https://ujhin.github.io/upbit-client-docs/
 
-- Upbit OPEN API Version: 1.2.0
+- Upbit OPEN API Version: 1.2.2
 - Author: ujhin
 - Email: [email protected]
 - GitHub: https://github.com/uJhin
@@ -28,8 +28,8 @@ def _get_versions(package_name):
 
 PACKAGE_NAME     = 'upbit-client'
 
-OPEN_API_VERSION = '1.2.0'
-CURRENT_VERSION  = OPEN_API_VERSION+'.4'
+OPEN_API_VERSION = '1.2.2'
+CURRENT_VERSION  = OPEN_API_VERSION+'.1'
 
 RELEASED_VERSION = _get_versions(PACKAGE_NAME)
 LATEST_VERSION   = RELEASED_VERSION[0]

upbit/websocket.py

@@ -75,6 +75,15 @@ def connect(
             ping_timeout=ping_timeout
         )
 
+    async def ping(self):
+        """
+        Client to Server PING
+        """
+        async with self as conn:
+            await conn.send('PING')
+            recv = await conn.recv()
+            return json.loads(recv)
+
     @staticmethod
     def generate_orderbook_codes(
         currencies: Union[List[str]],
@@ -163,12 +172,12 @@ def generate_payload(
 
         return json.dumps(payload)
 
-    async def __aenter__(self) -> websockets.client.WebSocketClientProtocol:
+    async def __aenter__(self):
         return await self.Connection.__aenter__()
 
-    async def __aexit__(self, exc_type, exc_value, traceback) -> None:
+    async def __aexit__(self, exc_type, exc_value, traceback):
         await self.Connection.__aexit__(exc_type, exc_value, traceback)
-    
+
     def __str__(self):
         return self.__repr__()