pikers
/
piker
3
0
Fork 0
Code Issues 2 Pull Requests 4 Packages Projects Releases Wiki Activity

Ensure we're passing the correct api version to the header builder, 

make headers a default arg
small_kucoin_fixes
jaredgoldman 2023-03-11 16:21:42 -05:00
parent cda045f123
commit 1a655b7e39
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
piker/brokers/kucoin.py
View File

@ -15,10 +15,9 @@
# along with this program. If not, see <https://www.gnu.org/licenses/>. # along with this program. If not, see <https://www.gnu.org/licenses/>.
""" """
Kucoin broker backend
""" """
from logging import warning
from typing import Any, Optional, Literal from typing import Any, Optional, Literal
from contextlib import asynccontextmanager as acm from contextlib import asynccontextmanager as acm
from datetime import datetime from datetime import datetime
@ -41,10 +40,6 @@ from piker.log import get_logger
from ._util import DataUnavailable from ._util import DataUnavailable
from piker.pp import config from piker.pp import config
_spawn_kwargs = {
"infect_asyncio": True,
}
log = get_logger(__name__) log = get_logger(__name__)
_ohlc_dtype = [ _ohlc_dtype = [
("index", int), ("index", int),
@ -68,7 +63,7 @@ class Client:
self._authenticated: bool = False self._authenticated: bool = False
config = get_config() config = get_config()
breakpoint()
if ("key_id" in config) and \ if ("key_id" in config) and \
("key_secret" in config) and \ ("key_secret" in config) and \
("key_passphrase" in config): ("key_passphrase" in config):
@ -83,11 +78,13 @@ class Client:
endpoint: str, endpoint: str,
api_v: str = "v2", api_v: str = "v2",
): ):
'''
https://docs.kucoin.com/#authentication
'''
now = int(time.time() * 1000) now = int(time.time() * 1000)
path = f'/api/{api_v}{endpoint}' path = f'/api/{api_v}{endpoint}'
str_to_sign = str(now) + action + path str_to_sign = str(now) + action + path
# Add headers to request if authenticated
signature = base64.b64encode( signature = base64.b64encode(
hmac.new( hmac.new(
self._key_secret.encode('utf-8'), self._key_secret.encode('utf-8'),
@ -109,7 +106,7 @@ class Client:
"KC-API-TIMESTAMP": str(now), "KC-API-TIMESTAMP": str(now),
"KC-API-KEY": self._key_id, "KC-API-KEY": self._key_id,
"KC-API-PASSPHRASE": passphrase, "KC-API-PASSPHRASE": passphrase,
"KC-API-KEY-VERSION": "2" "KC-API-KEY-VERSION": api_v[1]
} }
async def _request( async def _request(
@ -117,8 +114,8 @@ class Client:
action: Literal["POST", "GET", "PUT", "DELETE"], action: Literal["POST", "GET", "PUT", "DELETE"],
endpoint: str, endpoint: str,
api_v: str = "v2", api_v: str = "v2",
headers: dict = {}
) -> Any: ) -> Any:
headers = {}
if self._authenticated: if self._authenticated:
headers = self._gen_auth_req_headers(action, endpoint, api_v) headers = self._gen_auth_req_headers(action, endpoint, api_v)