recruiting-management 定制

yaml/replace.py

@@ -58,25 +58,37 @@ def _replayHanlder(rule, to, matched):
   return to;
 
 # 替换文件中的内容
-for file in files:
-  logger.info('开始替换文件:' + str(file))
-  for line in fileinput.input(file, inplace = 1):
-    for item in config:
-      rule = copy.deepcopy(item[1])
-      if rule.get('useNamespace', None) and rule['to']: 
-        rule['to'] = '-' + namespace + rule['to']
+if  project == 'recruiting-management':
+  for file in files:
+    logger.info('开始替换文件:' + str(file))
+    for line in fileinput.input(file, inplace = 1):
+      p = re.compile(r'' + 'api.stantoo.com')
+      to = 'recruitapi-' + namespace + '.liangkebang.net'
+      _handler1 = partial(_replayHanlder, 'api.stantoo.com', to)
+      line = re.sub(p, _handler1, line)
+      print(line, end = '')
+    fileinput.close()
 
-      if rule.get('useHost', None):
-        domain = sys.argv[1].split('.')[0]
-        rule['to'] = '//' + domain + rule['to']
+else:
+  for file in files:
+    logger.info('开始替换文件:' + str(file))
+    for line in fileinput.input(file, inplace = 1):
+      for item in config:
+        rule = copy.deepcopy(item[1])
+        if rule.get('useNamespace', None) and rule['to']: 
+          rule['to'] = '-' + namespace + rule['to']
 
-      if rule['reg']:
-        p = re.compile(r'' + rule['reg'])
-        _handler1 = partial(_replayHanlder, rule['reg'], rule['to'])
-        line = re.sub(p, _handler1, line)
+        if rule.get('useHost', None):
+          domain = sys.argv[1].split('.')[0]
+          rule['to'] = '//' + domain + rule['to']
 
-    print(line, end = '')
-  fileinput.close()
+        if rule['reg']:
+          p = re.compile(r'' + rule['reg'])
+          _handler1 = partial(_replayHanlder, rule['reg'], rule['to'])
+          line = re.sub(p, _handler1, line)
+
+      print(line, end = '')
+    fileinput.close()
 
 logger.info('完成')