mix
/
kea_for_miners
1
0
Fork 0
Code Issues 0 Pull Requests 0 Projects 0 Releases 0 Wiki Activity
Browse Source

Added REALLY work directory

master
root 7 years ago
parent
a81be736b8
commit
4759d3f703
3 changed files with 199 additions and 196 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +193
    -190
      kea_hosts_generane.py
  2. +0
    -1
      main.tpl
  3. +6
    -5
      web-leases.py

+ 193
- 190
kea_hosts_generane.py View File

@ -1,190 +1,193 @@
#!/usr/bin/python3
# coding=utf-8
# import os
# import subprocess
# import sys
import time
import paramiko
import MySQLdb
sshuser = 'root'
sshpass = 'admin'
dbhost = '10.5.1.248'
dbuser = 'kea'
dbpass = 'kea1234'
dbbase = 'dhcp4'
ippool = [10, 210]
q1c = "SELECT REPLACE(hw_addr, ':','') as hw_addr, " \
"serial, " \
"net_num " \
"FROM dhcp4.1c_import " \
"where imported = 0;"
qleases4 = "SELECT INET_NTOA(address) as ip4, " \
"HEX(hwaddr)as hw_addr, hostname, " \
"expire, " \
"valid_lifetime " \
"FROM lease4 " \
"where INET_NTOA(address) like '10.6.0%';"
qhosts_ip = "SELECT " \
"INET_NTOA(ipv4_address) AS ipv4_address " \
"FROM dhcp4.hosts;"
qinsert_hosts = "START TRANSACTION; " \
"SET @ipv4_reservation='%s'; " \
"SET @hostname = '%s'; " \
"SET @identifier_type='hw-address'; " \
"SET @identifier_value='%s'; " \
"SET @dhcp4_subnet_id=1024; " \
"INSERT INTO hosts ( dhcp_identifier, " \
" dhcp_identifier_type, " \
" dhcp4_subnet_id, " \
" ipv4_address, " \
" hostname) " \
"VALUES ( " \
"UNHEX(@identifier_value)," \
" (SELECT type FROM host_identifier_type WHERE name=@identifier_type)," \
" @dhcp4_subnet_id," \
" INET_ATON(@ipv4_reservation)," \
" @hostname); " \
"COMMIT;"
qupdate_1c = "update 1c_import set imported = 1 where serial = '%s';"
def send_data(user, password, name, host):
errorcode = ''
work = True
commands = []
# commands.append("hostname %s" % name)
# commands.append("echo '%s' > /etc/hostname" % name)
# commands.append("echo 'hostname=%s' > /etc/network.conf.factory" % name)
# commands.append("echo 'dhcp=true' >> /etc/network.conf.factory")
# commands.append("echo 'hostname=%s' > /config/network.conf" % name)
# commands.append("echo 'dhcp=true' >> /config/network.conf")
# commands.append("sync")
commands.append("reboot")
print(commands)
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) # если нет хоста - добавляем
try:
client.connect(hostname=host, username=user, password=password, look_for_keys=False,
allow_agent=False) # пытаемся подключиться, look_for_keys=False, allow_agent=False - зачем-то
# нужны циске.
except Exception as exept: # в случае исключения -
errorcode = exept # возвращаем код и описание исключения
else: # усли все в порядке
conn = client.invoke_shell() # врубаем интерактивный шелл - с цисками только так.
time.sleep(1) # задержка
conn.recv(1000) # хаваем все, что выплюнула, дабы не отсвечивало (ограничиваю 1000 строк)
conn.send("head -n 1 /config/network.conf\n")
time.sleep(1)
h = conn.recv(100)
if "hostname=" + name in h:
work = False
work = True # ЗАГЛУШКА!!! В данном случае условия не проверяются.
if work:
print(name, host, "will reboot")
for command in commands: # Читаем и исполняем команды
conn.send(command + '\n') # засылаем команду
time.sleep(0.1) # ждем указанный таймаут
conn.recv(99999999999) # хаваем его в никуда
else:
print("nothing to do")
finally: # по окончании всех процедур
client.close() # закрываем коннектий
return errorcode # Вертаем все, что наковыряли
def read_table(sql):
try:
conn = MySQLdb.connect(host=dbhost, user=dbuser,
passwd=dbpass, db=dbbase)
except MySQLdb.Error as err:
print("Connection error: {}".format(err))
conn.close()
try:
cur = conn.cursor(MySQLdb.cursors.DictCursor)
cur.execute(sql)
data = cur.fetchall()
except MySQLdb.Error as err:
print("Query error: {}".format(err))
conn.close()
return data
def first_free_ip(ips, ip_pool, net_num):
locked_ip = []
octet1 = ''
octet2 = ''
octet3 = net_num
octet4 = ''
set_octets = False
for ip in ips:
octets = ip['ipv4_address'].split('.')
if not set_octets:
octet1 = octets[0]
octet2 = octets[1]
set_octets = True
if octets[2] == str(net_num):
locked_ip.append(octets[3])
i = ip_pool[0]
while i < ip_pool[1]:
if not str(i) in locked_ip:
octet4 = str(i)
break
i = i + 1
if octet4 == '':
return ''
else:
return "%s.%s.%s.%s" % (octet1, octet2, octet3, octet4)
def make_hosts(sql, ip, hostname, hw_addr):
try:
conn = MySQLdb.connect(host=dbhost, user=dbuser,
passwd=dbpass, db=dbbase)
except MySQLdb.Error as err:
print("Connection error: {}".format(err))
conn.close()
try:
cur = conn.cursor(MySQLdb.cursors.DictCursor)
cur.execute(sql % (ip, hostname, hw_addr))
data = cur.fetchall()
cur.close()
except MySQLdb.Error as err:
print("Query error: {}".format(err))
try:
cur = conn.cursor(MySQLdb.cursors.DictCursor)
print(qupdate_1c % hostname)
cur.execute(qupdate_1c % hostname)
data = cur.fetchall()
cur.close()
except MySQLdb.Error as err:
print("Query error: {}".format(err))
conn.commit()
conn.close()
return data
def main():
data_1c = read_table(q1c)
data_leases = read_table(qleases4)
for line in data_1c:
set_ip = first_free_ip(read_table(qhosts_ip), ippool, line['net_num'])
set_hw_addr = line['hw_addr']
set_host = line['serial']
make_hosts(qinsert_hosts, set_ip, set_host, set_hw_addr)
print('host built name %s, mac %s, ip %s' % (set_host, set_hw_addr, set_ip))
for line in data_1c:
for lease in data_leases:
if line['hw_addr'] == lease['hw_addr']:
send_data(sshuser, sshpass, line['serial'], lease['ip4'])
if __name__ == '__main__':
main()
#! /usr/bin/python3
# coding=utf-8
# import os
# import subprocess
# import sys
import time
import paramiko
import MySQLdb
ip_oct1 = '10'
ip_oct2 = '5'
sshuser = 'root'
sshpass = 'admin'
dbhost = '10.5.1.248'
dbuser = 'kea'
dbpass = 'kea1234'
dbbase = 'dhcp4'
ippool = [10, 210]
q1c = "SELECT REPLACE(hw_addr, ':','') as hw_addr, " \
"serial, " \
"net_num " \
"FROM dhcp4.1c_import " \
"where imported = 0;"
qleases4 = "SELECT INET_NTOA(address) as ip4, " \
"HEX(hwaddr)as hw_addr, hostname, " \
"expire, " \
"valid_lifetime " \
"FROM lease4 " \
"where INET_NTOA(address) like '10.5.0%';"
qhosts_ip = "SELECT " \
"INET_NTOA(ipv4_address) AS ipv4_address " \
"FROM dhcp4.hosts;"
qinsert_hosts = "START TRANSACTION; " \
"SET @ipv4_reservation='%s'; " \
"SET @hostname = '%s'; " \
"SET @identifier_type='hw-address'; " \
"SET @identifier_value='%s'; " \
"SET @dhcp4_subnet_id=1024; " \
"INSERT INTO hosts ( dhcp_identifier, " \
" dhcp_identifier_type, " \
" dhcp4_subnet_id, " \
" ipv4_address, " \
" hostname) " \
"VALUES ( " \
"UNHEX(@identifier_value)," \
" (SELECT type FROM host_identifier_type WHERE name=@identifier_type)," \
" @dhcp4_subnet_id," \
" INET_ATON(@ipv4_reservation)," \
" @hostname); " \
"COMMIT;"
qupdate_1c = "update 1c_import set imported = 1 where serial = '%s';"
def send_data(user, password, name, host):
errorcode = ''
work = True
commands = []
# commands.append("hostname %s" % name)
# commands.append("echo '%s' > /etc/hostname" % name)
# commands.append("echo 'hostname=%s' > /etc/network.conf.factory" % name)
# commands.append("echo 'dhcp=true' >> /etc/network.conf.factory")
# commands.append("echo 'hostname=%s' > /config/network.conf" % name)
# commands.append("echo 'dhcp=true' >> /config/network.conf")
# commands.append("sync")
commands.append("reboot \n")
print(commands)
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) # если нет хоста - добавляем
try:
client.connect(hostname=host, username=user, password=password, look_for_keys=False,
allow_agent=False) # пытаемся подключиться, look_for_keys=False, allow_agent=False - зачем-то
# нужны циске.
except Exception as exept: # в случае исключения -
errorcode = exept # возвращаем код и описание исключения
else: # усли все в порядке
conn = client.invoke_shell() # врубаем интерактивный шелл - с цисками только так.
time.sleep(1) # задержка
conn.recv(1000) # хаваем все, что выплюнула, дабы не отсвечивало (ограничиваю 1000 строк)
# conn.send("head -n 1 /config/network.conf\n")
# time.sleep(1)
# h = conn.recv(100)
# if "hostname=" + name in h:
# work = False
work = True # ЗАГЛУШКА!!! В данном случае условия не проверяются.
if work:
print(name, host, "will reboot")
for command in commands: # Читаем и исполняем команды
conn.send(command + '\n') # засылаем команду
time.sleep(0.1) # ждем указанный таймаут
conn.recv(99999999999) # хаваем его в никуда
else:
print("nothing to do")
finally: # по окончании всех процедур
client.close() # закрываем коннектий
return errorcode # Вертаем все, что наковыряли
def read_table(sql):
try:
conn = MySQLdb.connect(host=dbhost, user=dbuser,
passwd=dbpass, db=dbbase)
except MySQLdb.Error as err:
print("Connection error: {}".format(err))
conn.close()
try:
cur = conn.cursor(MySQLdb.cursors.DictCursor)
cur.execute(sql)
data = cur.fetchall()
except MySQLdb.Error as err:
print("Query error: {}".format(err))
conn.close()
return data
def first_free_ip(ips, ip_pool, net_num):
locked_ip = []
octet1 = ip_oct1
octet2 = ip_oct2
octet3 = net_num
octet4 = ''
set_octets = False
for ip in ips:
octets = ip['ipv4_address'].split('.')
if not set_octets:
octet1 = octets[0]
octet2 = octets[1]
set_octets = True
if octets[2] == str(net_num):
locked_ip.append(octets[3])
i = ip_pool[0]
while i < ip_pool[1]:
if not str(i) in locked_ip:
octet4 = str(i)
break
i = i + 1
if octet4 == '':
return ''
else:
return "%s.%s.%s.%s" % (octet1, octet2, octet3, octet4)
def make_hosts(sql, ip, hostname, hw_addr):
try:
conn = MySQLdb.connect(host=dbhost, user=dbuser,
passwd=dbpass, db=dbbase)
except MySQLdb.Error as err:
print("Connection error: {}".format(err))
conn.close()
try:
cur = conn.cursor(MySQLdb.cursors.DictCursor)
cur.execute(sql % (ip, hostname, hw_addr))
data = cur.fetchall()
cur.close()
except MySQLdb.Error as err:
print("Query error: {}".format(err))
try:
cur = conn.cursor(MySQLdb.cursors.DictCursor)
print(qupdate_1c % hostname)
cur.execute(qupdate_1c % hostname)
data = cur.fetchall()
cur.close()
except MySQLdb.Error as err:
print("Query error: {}".format(err))
conn.commit()
conn.close()
return data
def main():
data_1c = read_table(q1c)
data_leases = read_table(qleases4)
for line in data_1c:
set_ip = first_free_ip(read_table(qhosts_ip), ippool, line['net_num'])
set_hw_addr = line['hw_addr']
set_host = line['serial']
make_hosts(qinsert_hosts, set_ip, set_host, set_hw_addr)
print('host built name %s, mac %s, ip %s' % (set_host, set_hw_addr, set_ip))
for line in data_1c:
for lease in data_leases:
if line['hw_addr'] == lease['hw_addr']:
send_data(sshuser, sshpass, line['serial'], lease['ip4'])
if __name__ == '__main__':
main()

+ 0
- 1
main.tpl View File

@ -2,7 +2,6 @@
<html lang="en">
<head>
<title>My Bottle Home</title>
<link rel="stylesheet" type="text/css" href="http://twitter.github.com/bootstrap/assets/css/bootstrap.css">
</head>
<body>
<h1>Initial page</h1>


+ 6
- 5
web-leases.py View File

@ -1,3 +1,6 @@
#!/usr/bin/python3
# coding=utf-8
from bottle import route, run, static_file, template
import MySQLdb
@ -19,7 +22,7 @@ def server_homepage():
" HEX(hwaddr)as hw_addr, " \
" hostname " \
" FROM lease4 " \
"where INET_NTOA(address) like '10.6.0%'" \
"where INET_NTOA(address) like '10.5.0%'" \
" ORDER BY expire DESC;"
try:
conn = MySQLdb.connect(host=dbhost, user=dbuser,
@ -34,16 +37,14 @@ def server_homepage():
except MySQLdb.Error as err:
print("Query error: {}".format(err))
conn.close()
print(data)
data_list = []
for row in data:
temp = []
temp.append(row['expire'])
temp.append(row['ip4'])
temp.append(row['hw_addr'])
temp.append(row['hostnamea'])
temp.append(row['hostname'])
data_list.append(temp)
print(data_list)
return template('main.tpl', A = data_list)
run(host=HOST, port=8080, debug=True)
run(host='10.5.1.248', port=8080, debug=True)

Write Preview
Loading…
Cancel
Save

Powered by TurnKey Linux.