public upgrade py

tools/upgrade/opts.py

@@ -0,0 +1,193 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from my_error import MyError
+import sys
+import os
+import getopt
+
+help_str = \
+"""
+Help:
+""" +\
+sys.argv[0] + """ [OPTIONS]""" +\
+'\n\n' +\
+'-I, --help          Display this help and exit.\n' +\
+'-V, --version       Output version information and exit.\n' +\
+'-h, --host=name     Connect to host.\n' +\
+'-P, --port=name     Port number to use for connection.\n' +\
+'-u, --user=name     User for login.\n' +\
+'-p, --password=name Password to use when connecting to server. If password is\n' +\
+'                    not given it\'s empty string "".\n' +\
+'-m, --module=name   Modules to run. Modules should be a string combined by some of\n' +\
+'                    the following strings: ddl, normal_dml, each_tenant_dml,\n' +\
+'                    system_variable_dml, special_action, all. "all" represents\n' +\
+'                    that all modules should be run. They are splitted by ",".\n' +\
+'                    For example: -m all, or --module=ddl,normal_dml,special_action\n' +\
+'-l, --log-file=name Log file path. If log file path is not given it\'s ' + os.path.splitext(sys.argv[0])[0] + '.log\n' +\
+'\n\n' +\
+'Maybe you want to run cmd like that:\n' +\
+sys.argv[0] + ' -h 127.0.0.1 -P 3306 -u admin -p admin\n'
+
+version_str = """version 1.0.0"""
+
+class Option:
+  __g_short_name_set = set([])
+  __g_long_name_set = set([])
+  __short_name = None
+  __long_name = None
+  __is_with_param = None
+  __is_local_opt = None
+  __has_value = None
+  __value = None
+  def __init__(self, short_name, long_name, is_with_param, is_local_opt, default_value = None):
+    if short_name in Option.__g_short_name_set:
+      raise MyError('duplicate option short name: {0}'.format(short_name))
+    elif long_name in Option.__g_long_name_set:
+      raise MyError('duplicate option long name: {0}'.format(long_name))
+    Option.__g_short_name_set.add(short_name)
+    Option.__g_long_name_set.add(long_name)
+    self.__short_name = short_name
+    self.__long_name = long_name
+    self.__is_with_param = is_with_param
+    self.__is_local_opt = is_local_opt
+    self.__has_value = False
+    if None != default_value:
+      self.set_value(default_value)
+  def is_with_param(self):
+    return self.__is_with_param
+  def get_short_name(self):
+    return self.__short_name
+  def get_long_name(self):
+    return self.__long_name
+  def has_value(self):
+    return self.__has_value
+  def get_value(self):
+    return self.__value
+  def set_value(self, value):
+    self.__value = value
+    self.__has_value = True
+  def is_local_opt(self):
+    return self.__is_local_opt
+  def is_valid(self):
+    return None != self.__short_name and None != self.__long_name and True == self.__has_value and None != self.__value
+
+g_opts =\
+[\
+Option('I', 'help', False, True),\
+Option('V', 'version', False, True),\
+Option('h', 'host', True, False),\
+Option('P', 'port', True, False),\
+Option('u', 'user', True, False),\
+Option('p', 'password', True, False, ''),\
+# 要跑哪个模块，默认全跑
+Option('m', 'module', True, False, 'all'),\
+# 日志文件路径，不同脚本的main函数中中会改成不同的默认值
+Option('l', 'log-file', True, False)
+]\
+
+def change_opt_defult_value(opt_long_name, opt_default_val):
+  global g_opts
+  for opt in g_opts:
+    if opt.get_long_name() == opt_long_name:
+      opt.set_value(opt_default_val)
+      return
+
+def has_no_local_opts():
+  global g_opts
+  no_local_opts = True
+  for opt in g_opts:
+    if opt.is_local_opt() and opt.has_value():
+      no_local_opts = False
+  return no_local_opts
+
+def check_db_client_opts():
+  global g_opts
+  for opt in g_opts:
+    if not opt.is_local_opt() and not opt.has_value():
+      raise MyError('option "-{0}" has not been specified, maybe you should run "{1} --help" for help'\
+          .format(opt.get_short_name(), sys.argv[0]))
+
+def parse_option(opt_name, opt_val):
+  global g_opts
+  for opt in g_opts:
+    if opt_name in (('-' + opt.get_short_name()), ('--' + opt.get_long_name())):
+      opt.set_value(opt_val)
+
+def parse_options(argv):
+  global g_opts
+  short_opt_str = ''
+  long_opt_list = []
+  for opt in g_opts:
+    if opt.is_with_param():
+      short_opt_str += opt.get_short_name() + ':'
+    else:
+      short_opt_str += opt.get_short_name()
+  for opt in g_opts:
+    if opt.is_with_param():
+      long_opt_list.append(opt.get_long_name() + '=')
+    else:
+      long_opt_list.append(opt.get_long_name())
+  (opts, args) = getopt.getopt(argv, short_opt_str, long_opt_list)
+  for (opt_name, opt_val) in opts:
+    parse_option(opt_name, opt_val)
+  if has_no_local_opts():
+    check_db_client_opts()
+
+def deal_with_local_opt(opt):
+  if 'help' == opt.get_long_name():
+    global help_str
+    print help_str
+  elif 'version' == opt.get_long_name():
+    global version_str
+    print version_str
+
+def deal_with_local_opts():
+  global g_opts
+  if has_no_local_opts():
+    raise MyError('no local options, can not deal with local options')
+  else:
+    for opt in g_opts:
+      if opt.is_local_opt() and opt.has_value():
+        deal_with_local_opt(opt)
+        # 只处理一个
+        return
+
+def get_opt_host():
+  global g_opts
+  for opt in g_opts:
+    if 'host' == opt.get_long_name():
+      return opt.get_value()
+
+def get_opt_port():
+  global g_opts
+  for opt in g_opts:
+    if 'port' == opt.get_long_name():
+      return opt.get_value()
+
+def get_opt_user():
+  global g_opts
+  for opt in g_opts:
+    if 'user' == opt.get_long_name():
+      return opt.get_value()
+
+def get_opt_password():
+  global g_opts
+  for opt in g_opts:
+    if 'password' == opt.get_long_name():
+      return opt.get_value()
+
+def get_opt_module():
+  global g_opts
+  for opt in g_opts:
+    if 'module' == opt.get_long_name():
+      return opt.get_value()
+
+def get_opt_log_file():
+  global g_opts
+  for opt in g_opts:
+    if 'log-file' == opt.get_long_name():
+      return opt.get_value()
+
+#parse_options(sys.argv[1:])
+