ahmedbilal/etcd3_wrapper
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

if arg of put is str don't decode it

Browse source
This commit is contained in:
ahmadbilalkhalid 2019-06-29 17:14:13 +05:00
parent deee2b59cb
commit 96dc46032c
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
etcd_wrapper.py
View file

@ -2,6 +2,7 @@ import etcd3
import json import json
from dataclasses import dataclass from dataclasses import dataclass
@dataclass(init=False) @dataclass(init=False)
class EtcdEntry: class EtcdEntry:
def __init__(self, meta, value, value_in_json=False): def __init__(self, meta, value, value_in_json=False):
@ -15,6 +16,7 @@ class EtcdEntry:
key: str key: str
value: str value: str
class Etcd3Wrapper(object): class Etcd3Wrapper(object):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
self.client = etcd3.client(*args, **kwargs) self.client = etcd3.client(*args, **kwargs)
@ -29,8 +31,11 @@ class Etcd3Wrapper(object):
_key, _value = args _key, _value = args
if value_in_json: if value_in_json:
_value = json.dumps(_value) _value = json.dumps(_value)
return self.client.put(_key.decode("utf-8"), _value)
if not isinstance(_key, str):
_key = _key.decode("utf-8")
return self.client.put(_key, _value, **kwargs)
def get_prefix(self, *args, value_in_json=False, **kwargs): def get_prefix(self, *args, value_in_json=False, **kwargs):
r = self.client.get_prefix(*args, **kwargs) r = self.client.get_prefix(*args, **kwargs)