#!/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:

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.")