Как назначить другое значение переменной в производственном скрипте и тестовом скрипте,

Question

эта функция создает каталог, мы зависим от режима, чтобы определить, является ли сценарий рабочим кодом или тестовым файлом. При создании тестового файла я def setup ():

Flags.is_test = True ( Это также не должно использоваться * )

Есть подход: Исключить все ссылки на FLAGS.is_test. Вместо, # определить режим как атрибут модуля и сделать эти функции классовыми которые используют атрибут модуля. Позже, когда вы создаете приложение, используя этот модуль, то если производство, введите значение 1111 для режима. Если развитие, то залить значение 1234.

_CreateDirectory(path, is_dir=False):

  if FLAGS.is_test:
    mode = 1234
  else:
    mode = 1111

  if '/private_file_location/' in path:
    if not Modulefile.Exists(directory_name):
      _CreateDirectory(directory_name, True)
    if not file.Exists(path) and is_dir:
      Modulefile.MkDir(path, mode=mode)
      Moudlefile.SetMode(path, mode)
  else:
    if not os.path.exists(directory_name):
      _CreateDirectory(directory_name, True)
    if not os.path.exists(path) and is_dir:
      os.mkdir(path, mode)
      os.chmod(path, mode)

Может кто-нибудь переписать сценарий для меня. Спасибо