multi-server support with HashClient (#1)

Author: skywaymsn

flask_pymemcache.py

@@ -57,6 +57,7 @@
 from __future__ import absolute_import, division, print_function
 import flask
 import pymemcache.client
+import pymemcache.client.hash
 
 
 class FlaskPyMemcache(object):
@@ -82,7 +83,15 @@ def init_app(self, app, conf_key=None):
             raise TypeError("Flask-PyMemcache conf should be dict")
 
         close_on_teardown = conf.pop('close_on_teardown', False)
-        client = pymemcache.client.Client(**conf)
+
+        if isinstance(conf['server'], list):
+            conf['servers'] = conf.pop('server')
+            client = pymemcache.client.hash.HashClient(**conf)
+        elif isinstance(conf['server'], tuple):
+            client = pymemcache.client.Client(**conf)
+        else:
+            raise TypeError("Flask-PyMemcache conf['server'] should be tuple or list of tuples")
+
         app.extensions.setdefault('pymemcache', {})
         app.extensions['pymemcache'][self] = client
 

setup.py

@@ -19,7 +19,7 @@
     url='https://github.com/KLab/Flask-PyMemcache',
     description="pymemcache integration for Flask",
     long_description=readme,
-    install_requires=["Flask", "pymemcache>=1.2.3", "six"],
+    install_requires=["Flask", "pymemcache>=1.3.4", "six"],
     test_require=['pytest'],
     test_suite='test_flask_pymemcache',
 )