add

trans2es/management/commands/trans2es_tipsdata2es.py

@@ -11,30 +11,29 @@ import sys
 
 from libs.es import ESPerform
 import trans2es.models as md
-from libs.table_scan import TableSlicer,TableSlicerChunk
-from trans2es.type_info import get_type_info_map,TypeInfo,tips_attr_sync_to_redis_type_info_map
+from libs.table_scan import TableSlicer, TableSlicerChunk
+from trans2es.type_info import get_type_info_map, TypeInfo, tips_attr_sync_to_redis_type_info_map
 
 from libs.cache import redis_client
 import json
 
 
-
-
 class Command(BaseCommand):
-
     args = ''
     help = 'dump data to redis, parallel'
 
     from optparse import make_option
 
     option_list = BaseCommand.option_list + (
-        make_option('-t', '--type', dest='type', help='type name to dump data to elasticsearch', metavar='TYPE',default=''),
-        make_option('-i', '--index-prefix', dest='index_prefix', help='index name to dump data to elasticsearch', metavar='INDEX_PREFIX'),
+        make_option('-t', '--type', dest='type', help='type name to dump data to elasticsearch', metavar='TYPE',
+                    default=''),
+        make_option('-i', '--index-prefix', dest='index_prefix', help='index name to dump data to elasticsearch',
+                    metavar='INDEX_PREFIX'),
         make_option('-p', '--parallel', dest='parallel', help='parallel process count', metavar='PARALLEL'),
         make_option('-s', '--pks', dest='pks', help='specify sync pks, comma separated', metavar='PKS', default=''),
         make_option('--streaming-slicing', dest='streaming_slicing', action='store_true', default=True),
         make_option('--no-streaming-slicing', dest='streaming_slicing', action='store_false', default=True),
-        make_option('-S', '--sync_type',dest='sync_type', help='sync data to es',metavar='TYPE',default='')
+        make_option('-S', '--sync_type', dest='sync_type', help='sync data to es', metavar='TYPE', default='')
     )
 
     def handle(self, *args, **options):
@@ -43,7 +42,7 @@ class Command(BaseCommand):
             for type_name in type_name_list:
 
                 if len(options["type"]):
-                    if options["type"] == "all" or type_name==options["type"]:
+                    if options["type"] == "all" or type_name == options["type"]:
                         logging.info("begin sync [%s] data to redis!" % type_name)
                         type_info = tips_attr_sync_to_redis_type_info_map()[type_name]
                         query_set = type_info.queryset