75 lines
2.0 KiB
Python
Executable File
75 lines
2.0 KiB
Python
Executable File
# coding: utf-8
|
||
|
||
import sys
|
||
import os
|
||
import json
|
||
import time
|
||
|
||
sys.path.append(os.getcwd() + "/class/core")
|
||
import mw
|
||
import db
|
||
|
||
# cmd = 'ls /usr/local/lib/ | grep python | cut -d \\ -f 1 | awk \'END {print}\''
|
||
# info = mw.execShell(cmd)
|
||
# p = "/usr/local/lib/" + info[0].strip() + "/site-packages"
|
||
# sys.path.append(p)
|
||
|
||
|
||
def set_panel_pwd(password, ncli=False):
|
||
# 设置面板密码
|
||
import db
|
||
sql = db.Sql()
|
||
result = sql.table('users').where('id=?', (1,)).setField(
|
||
'password', mw.md5(password))
|
||
username = sql.table('users').where('id=?', (1,)).getField('username')
|
||
if ncli:
|
||
print("|-用户名: " + username)
|
||
print("|-新密码: " + password)
|
||
else:
|
||
print(username)
|
||
|
||
|
||
def set_panel_username(username=None):
|
||
# 随机面板用户名
|
||
import db
|
||
sql = db.Sql()
|
||
if username:
|
||
if len(username) < 5:
|
||
print("|-错误，用户名长度不能少于5位")
|
||
return
|
||
if username in ['admin', 'root']:
|
||
print("|-错误，不能使用过于简单的用户名")
|
||
return
|
||
|
||
sql.table('users').where('id=?', (1,)).setField('username', username)
|
||
print("|-新用户名: %s" % username)
|
||
return
|
||
|
||
username = sql.table('users').where('id=?', (1,)).getField('username')
|
||
if username == 'admin':
|
||
username = mw.getRandomString(8).lower()
|
||
sql.table('users').where('id=?', (1,)).setField('username', username)
|
||
print('username: ' + username)
|
||
|
||
|
||
def getServerIp():
|
||
version = sys.argv[2]
|
||
ip = mw.execShell(
|
||
"curl -{} -sS --connect-timeout 5 -m 60 https://v6r.ipip.net/?format=text".format(version))
|
||
print(ip[0])
|
||
|
||
|
||
if __name__ == "__main__":
|
||
type = sys.argv[1]
|
||
if type == 'panel':
|
||
set_panel_pwd(sys.argv[2])
|
||
elif type == 'username':
|
||
if len(sys.argv) > 2:
|
||
set_panel_username(sys.argv[2])
|
||
else:
|
||
set_panel_username()
|
||
elif type == 'getServerIp':
|
||
getServerIp()
|
||
else:
|
||
print('ERROR: Parameter error')
|