/usr/share/pyshared/pylibmc/client.py is in python-pylibmc 1.2.2-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 | """Python-level wrapper client"""
import _pylibmc
from .consts import (hashers, distributions, all_behaviors,
hashers_rvs, distributions_rvs,
BehaviorDict)
_all_behaviors_set = set(all_behaviors)
server_type_map = {"tcp": _pylibmc.server_type_tcp,
"udp": _pylibmc.server_type_udp,
"unix": _pylibmc.server_type_unix}
def _split_spec_type(spec):
if spec.startswith("/"):
return ("unix", spec)
else:
if spec.startswith("udp:"):
return ("udp", spec[4:])
else:
return ("tcp", spec)
def _splitport(spec, port=None):
if spec.startswith("["):
if "]" not in spec:
raise ValueError(spec)
if spec.endswith("]"):
addr = spec[1:-1]
else:
(addr, port) = spec[1:].rsplit("]:", 1)
elif ":" in spec:
(addr, port) = spec.split(":", 1)
else:
addr = spec
return (addr, port)
def translate_server_spec(server, port=11211):
"""Translate/normalize specification *server* into three-tuple (tp, addr, port).
This is format is used by the extension module.
>>> translate_server_spec("127.0.0.1")
(1, '127.0.0.1', 11211)
>>> translate_server_spec("udp:127.0.0.1")
(2, '127.0.0.1', 11211)
>>> translate_server_spec("/var/run/memcached.sock")
(4, '/var/run/memcached.sock', 0)
>>> translate_server_spec("127.0.0.1:22122")
(1, '127.0.0.1', 22122)
>>> translate_server_spec("127.0.0.1", port=1234)
(1, '127.0.0.1', 1234)
>>> translate_server_spec("[::1]:22122")
(1, '::1', 22122)
>>> translate_server_spec("[::1]")
(1, '::1', 11211)
"""
(scheme, spec) = _split_spec_type(server)
stype = server_type_map[scheme]
if scheme == "unix":
(addr, port) = (spec, 0)
elif scheme in ("tcp", "udp"):
(addr, port) = _splitport(spec, port=port)
return (stype, str(addr), int(port))
def translate_server_specs(servers):
addr_tups = []
for server in servers:
# Anti-pattern for convenience
if isinstance(server, tuple) and len(server) == 3:
addr_tup = server
else:
addr_tup = translate_server_spec(server)
addr_tups.append(addr_tup)
return addr_tups
class Client(_pylibmc.client):
def __init__(self, servers, behaviors=None, binary=False,
username=None, password=None):
"""Initialize a memcached client instance.
This connects to the servers in *servers*, which will default to being
TCP servers. If it looks like a filesystem path, a UNIX socket. If
prefixed with `udp:`, a UDP connection.
If *binary* is True, the binary memcached protocol is used.
SASL authentication is supported if libmemcached supports it (check
*pylibmc.support_sasl*). Requires both username and password.
Note that SASL requires *binary*=True.
"""
self.binary = binary
self.addresses = list(servers)
super(Client, self).__init__(servers=translate_server_specs(servers),
binary=binary, username=username,
password=password)
if behaviors is not None:
self.set_behaviors(behaviors)
def __repr__(self):
return "%s(%r, binary=%r)" % (self.__class__.__name__,
self.addresses, self.binary)
def __str__(self):
addrs = ", ".join(map(str, self.addresses))
return "<%s for %s, binary=%r>" % (self.__class__.__name__,
addrs, self.binary)
# {{{ Mapping interface
def __getitem__(self, key):
value = self.get(key)
if value is None:
raise KeyError(key)
else:
return value
def __setitem__(self, key, value):
if not self.set(key, value):
raise KeyError("failed setting %r" % (key,))
def __delitem__(self, key):
if not self.delete(key):
raise KeyError(key)
def __contains__(self, key):
return self.get(key) is not None
# }}}
# {{{ Behaviors
def get_behaviors(self):
"""Gets the behaviors from the underlying C client instance.
Reverses the integer constants for `hash` and `distribution` into more
understandable string values. See *set_behaviors* for info.
"""
bvrs = super(Client, self).get_behaviors()
bvrs["hash"] = hashers_rvs[bvrs["hash"]]
bvrs["distribution"] = distributions_rvs[bvrs["distribution"]]
return BehaviorDict(self, bvrs)
def set_behaviors(self, behaviors):
"""Sets the behaviors on the underlying C client instance.
Takes care of morphing the `hash` key, if specified, into the
corresponding integer constant (which the C client expects.) If,
however, an unknown value is specified, it's passed on to the C client
(where it most surely will error out.)
This also happens for `distribution`.
"""
unknown = set(behaviors).difference(_all_behaviors_set)
if unknown:
names = ", ".join(map(str, sorted(unknown)))
raise ValueError("unknown behavior names: %s" % (names,))
behaviors = behaviors.copy()
if behaviors.get("hash") is not None:
behaviors["hash"] = hashers[behaviors["hash"]]
if behaviors.get("ketama_hash") in hashers:
behaviors["ketama_hash"] = hashers[behaviors["ketama_hash"]]
if behaviors.get("distribution") is not None:
behaviors["distribution"] = distributions[behaviors["distribution"]]
return super(Client, self).set_behaviors(behaviors)
behaviors = property(get_behaviors, set_behaviors)
@property
def behaviours(self):
raise AttributeError("nobody uses british spellings")
# }}}
def clone(self):
obj = super(Client, self).clone()
obj.addresses = list(self.addresses)
obj.binary = self.binary
return obj
if __name__ == "__main__":
import doctest
doctest.testmod()
|