1. 程式人生 > >簡單的堡壘機模型

簡單的堡壘機模型

python

剛學python,寫了一個簡單的堡壘機模型,功能自己添加


#!/usr/bin/env python

#coding:utf-8


import MySQLdb

import paramiko

#connect mysql

conn = MySQLdb.connect("218.197.119.18","zhangpan","845896787","baoleiji")

cur = conn.cursor()


sqlUser = "select * from User"

sqlHost = "select * from Host"


cur0 = cur.execute(sqlUser)


#get User rows


result1 = cur.fetchmany(cur0)


for row in result1:

check_name = row[1]

check_passwd = row[2]



host_list = []

#login

while 1:

username = raw_input(‘username:‘).strip()

password = raw_input(‘password:‘).strip()

if username==check_name and password == check_passwd:

cur1=cur.execute(sqlHost)

result2 = cur.fetchmany(cur1)


for item in result2: host_list.append(item)

print ‘all host list:\n‘

for item1 in host_list:

host_id = item1[0]

host_name = item1[3]

ip_address = item1[1]

port = item1[2]

print host_id,host_name,ip_address,port

while 1:

choose = raw_input("please input you want connect host:").strip()

user_choose = int(choose)

if user_choose<len(host_list):

hostip = host_list[user_choose][1]

username = host_list[user_choose][4]

passwd = host_list[user_choose][5]

port = int(host_list[user_choose][2])

print hostip,username,passwd,port

ssh = paramiko.SSHClient()

ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

ssh.connect(hostip,22,username,passwd)

stdin,stdout,stderr = ssh.exec_command(‘mkdir python_study‘)

ssh. close()

else:

print ‘input error host id !‘

else:

print ‘Your Username or password is error‘


本文出自 “塵” 博客,請務必保留此出處http://zhangpan.blog.51cto.com/2818271/1924323

簡單的堡壘機模型