73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: UTF-8 -*-
|
|
|
|
import http.client
|
|
import sys
|
|
import getopt as opts
|
|
import urllib.parse
|
|
import re
|
|
|
|
|
|
def get_params() -> dict:
|
|
params = {
|
|
'-h': '127.0.0.1',
|
|
'-p': '8848',
|
|
'-t': '',
|
|
'-g': 'SEATA_GROUP',
|
|
'-u': '',
|
|
'-w': ''
|
|
}
|
|
inputs, args = opts.getopt(sys.argv[1:], shortopts='h:p:t:g:u:w:')
|
|
for k, v in inputs:
|
|
params[k] = v
|
|
print(params)
|
|
return params
|
|
|
|
def error_exit():
|
|
print('python nacos-config.py [-h host] [-p port] [-t tenant] [-g group] [-u username] [-w password]')
|
|
exit()
|
|
|
|
def get_pair(line: str) -> tuple:
|
|
res = re.match(r"([\.\w]+)=(.*)",line)
|
|
return res.groups() if res is not None else ['','']
|
|
|
|
|
|
headers = {
|
|
'content-type': "application/x-www-form-urlencoded"
|
|
}
|
|
|
|
hasError = False
|
|
|
|
params = get_params()
|
|
|
|
url_prefix = f"{params['-h']}:{params['-p']}"
|
|
tenant = params['-t']
|
|
username = params['-u']
|
|
password = params['-w']
|
|
group = params['-g']
|
|
url_postfix_base = f'/nacos/v1/cs/configs?group={group}&tenant={tenant}'
|
|
|
|
if username != '' and password != '':
|
|
url_postfix_base += f'&username={username}&password={password}'
|
|
|
|
if url_prefix == ':':
|
|
error_exit()
|
|
|
|
for line in open('../config.txt'):
|
|
pair = get_pair(line.rstrip("\n"))
|
|
if len(pair) < 2 or pair[0] == '' or pair[0].startswith("#") or pair[1] == '':
|
|
continue
|
|
url_postfix = url_postfix_base + f'&dataId={urllib.parse.quote(str(pair[0]))}&content={urllib.parse.quote(str(pair[1])).strip()}'
|
|
conn = http.client.HTTPConnection(url_prefix)
|
|
conn.request("POST", url_postfix, headers=headers)
|
|
res = conn.getresponse()
|
|
data = res.read().decode("utf-8")
|
|
if data != "true":
|
|
hasError = True
|
|
print(f"{pair[0]}={pair[1]} {data if hasError else 'success'}")
|
|
|
|
if hasError:
|
|
print("init nacos config fail.")
|
|
else:
|
|
print("init nacos config finished, please start seata-server.")
|