chamnan · April 18, 2019 08:54 · Apr 18, 2019 · Mar 7, 2017 · Feb 27, 2014 · Feb 27, 2014
diff --git a/redis_migrate.py b/redis_migrate.py
@@ -1,47 +1,45 @@
 #!/usr/bin/env python
 import argparse
 import redis
+import ast
 
-
-def connect_redis(conn_dict):
+def connect_redis(conn_dict):    
     conn = redis.StrictRedis(host=conn_dict['host'],
                              port=conn_dict['port'],
-                             db=conn_dict['db'])
+                             db=int(conn_dict['db']),
+                             password=conn_dict['auth'])
     return conn
 
-
 def conn_string_type(string):
-    format = '<host>:<port>/<db>'
+    format = "{'host':'127.0.0.1', 'port':3679, 'db': 1, 'auth':'*Passwword*'}"
     try:
-        host, portdb = string.split(':')
-        port, db = portdb.split('/')
-        db = int(db)
-    except ValueError:
+        conn_dict = ast.literal_eval(string)
+        if conn_dict.keys() != {'host', 'port', 'db', 'auth'}:
+            raise argparse.ArgumentTypeError('incorrect format, should be: %s' % format)   
+    except:
         raise argparse.ArgumentTypeError('incorrect format, should be: %s' % format)
-    return {'host': host,
-            'port': port,
-            'db': db}
-
+    return conn_dict
 
 def migrate_redis(source, destination):
     src = connect_redis(source)
     dst = connect_redis(destination)
-    for key in src.keys('*'):
+    for key in src.keys('Testing:*'):
         ttl = src.ttl(key)
         # we handle TTL command returning -1 (no expire) or -2 (no key)
         if ttl < 0:
             ttl = 0
-        print "Dumping key: %s" % key
+        print("Dumping key: %s" % key)
         value = src.dump(key)
-        print "Restoring key: %s" % key
+        if value is None:
+            value = ''
+        print("Restoring key: %s" % key)
         try:
             dst.restore(key, ttl * 1000, value, replace=True)
         except redis.exceptions.ResponseError:
-            print "Failed to restore key: %s" % key
+            print("Failed to restore key: %s" % key)
             pass
     return
 
-
 def run():
     parser = argparse.ArgumentParser()
     parser.add_argument('source', type=conn_string_type)

diff --git a/redis_migrate.py b/redis_migrate.py
@@ -34,7 +34,11 @@ def migrate_redis(source, destination):
         print "Dumping key: %s" % key
         value = src.dump(key)
         print "Restoring key: %s" % key
-        dst.restore(key, ttl, value)
+        try:
+            dst.restore(key, ttl * 1000, value, replace=True)
+        except redis.exceptions.ResponseError:
+            print "Failed to restore key: %s" % key
+            pass
     return
 
 

diff --git a/redis_migrate.py b/redis_migrate.py
@@ -35,7 +35,6 @@ def migrate_redis(source, destination):
         value = src.dump(key)
         print "Restoring key: %s" % key
         dst.restore(key, ttl, value)
-        return
     return
 
 

diff --git a/redis_migrate.py b/redis_migrate.py
@@ -28,6 +28,9 @@ def migrate_redis(source, destination):
     dst = connect_redis(destination)
     for key in src.keys('*'):
         ttl = src.ttl(key)
+        # we handle TTL command returning -1 (no expire) or -2 (no key)
+        if ttl < 0:
+            ttl = 0
         print "Dumping key: %s" % key
         value = src.dump(key)
         print "Restoring key: %s" % key

diff --git a/redis_migrate.py b/redis_migrate.py
@@ -27,10 +27,11 @@ def migrate_redis(source, destination):
     src = connect_redis(source)
     dst = connect_redis(destination)
     for key in src.keys('*'):
+        ttl = src.ttl(key)
         print "Dumping key: %s" % key
         value = src.dump(key)
         print "Restoring key: %s" % key
-        dst.restore(key, 0, value)
+        dst.restore(key, ttl, value)
         return
     return
 

diff --git a/redis_migrate.py b/redis_migrate.py
@@ -26,8 +26,12 @@ def conn_string_type(string):
 def migrate_redis(source, destination):
     src = connect_redis(source)
     dst = connect_redis(destination)
-    print "test_key src", src.get('test_key')
-    print "test_key dst", dst.get('test_key')
+    for key in src.keys('*'):
+        print "Dumping key: %s" % key
+        value = src.dump(key)
+        print "Restoring key: %s" % key
+        dst.restore(key, 0, value)
+        return
     return
 
 

diff --git a/redis_migrate.py b/redis_migrate.py
@@ -0,0 +1,42 @@
+#!/usr/bin/env python
+import argparse
+import redis
+
+
+def connect_redis(conn_dict):
+    conn = redis.StrictRedis(host=conn_dict['host'],
+                             port=conn_dict['port'],
+                             db=conn_dict['db'])
+    return conn
+
+
+def conn_string_type(string):
+    format = '<host>:<port>/<db>'
+    try:
+        host, portdb = string.split(':')
+        port, db = portdb.split('/')
+        db = int(db)
+    except ValueError:
+        raise argparse.ArgumentTypeError('incorrect format, should be: %s' % format)
+    return {'host': host,
+            'port': port,
+            'db': db}
+
+
+def migrate_redis(source, destination):
+    src = connect_redis(source)
+    dst = connect_redis(destination)
+    print "test_key src", src.get('test_key')
+    print "test_key dst", dst.get('test_key')
+    return
+
+
+def run():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('source', type=conn_string_type)
+    parser.add_argument('destination', type=conn_string_type)
+    options = parser.parse_args()
+    migrate_redis(options.source, options.destination)
+
+if __name__ == '__main__':
+    run()