__author__ = "shikun"
class Const(object):
    log = "D:/app/Apache2.2/htdocs/client.log" # apache 的log路径
    APAHEC_IP = "192.168.1.38" # 本机ip
    PORT_NUMBER = 8088 # 端口号

from http.server import BaseHTTPRequestHandler,HTTPServer
import urllib.parse
from common import operateFile
from common.customConst import Const
class myHandler(BaseHTTPRequestHandler):
  # Handler for the GET requests
    def do_GET(self):
        print('Get request received')
        req = urllib.parse.unquote(self.path)
        result = urllib.parse.parse_qs(req[2:]) # 得到请求参数
        self.send_response(200)
        self.send_header('Content-type','text/html')
        self.end_headers()
        # Send the html message
        self.wfile.write(b"ok!") #发送信息给客户端
        operateFile.write_txt(line=result["msg"][0], f_path=Const.log) # 记录各个客户端发来的信息
try:
  server = HTTPServer((Const.APAHEC_IP, Const.PORT_NUMBER), myHandler)
  print ('Started httpserver on port ' , Const.PORT_NUMBER)
  server.serve_forever()
except KeyboardInterrupt:
  print ('^C received, shutting down the web server')
  server.socket.close()

from multiprocessing import Process
from gevent import monkey; monkey.patch_all()
import gevent
import requests
import json
from common.customConst import Const
url = "http://rap.taobao.org/mockjsdata/10296/getUserInfo?id=2"
num = 10
result = {"success": 0, "computer": "压测客户端1", "cpu": 4, "men": 4, sum: 10} # 客户端信息

class Producer(object):
    '''
    协程发请求
    '''
    def __init__(self):
       self._rungevent()
       self.h = 0
    def _rungevent(self):
        jobs = []
        for i in range(num): #windows下有端口限制
            jobs.append(gevent.spawn(self.produce))
        gevent.joinall(jobs)
        requests.get("http://"+Const.APAHEC_IP+":"+ str(Const.PORT_NUMBER)+"/?msg="+str(result)) # 发送客户端的请求情况
    def produce(self):
        r = requests.get(url)
        if r.status_code == 200:
            r.encoding = 'UTF-8'
            if json.loads(r.text)["code"] == 0:
                result["success"] += 1
        else:
            print("失败咯")
def main():
    p = Process(target=Producer, args=())  # 一个进程启动协程
    p.start()

    # p1 = Process(target = Producer, args=())
    # p1.start()

if __name__ == '__main__':
    main()

import threading
class base_thread(threading.Thread):
    def __init__(self, func):
        threading.Thread.__init__(self)
        self.func = func
        #print(type(self.func))
    def run(self):
        self.func

import urllib.request as request
from bs4 import BeautifulSoup
from Base.Threads import base_thread
path = 'e:/apps/pic/'
def CrawlerFunc(url):
    list_img = getUrl(url)
    multi_thread(len(list_img), downloadImg(list_img))
def getUrl(url):
    response = request.urlopen(url)
    html = response.read()
    data = html.decode('utf-8')
    soup = BeautifulSoup(data)
    list_img = []
    for list in soup.find_all("img", {"class", "BDE_Image"}):
        list_img.append(list.attrs["src"])
    return list_img

def downloadImg(list_img):
    count = 1
    for i in list_img:
        filepath = path + str(count)+".jpg"
        with open(filepath, 'wb') as file:
            print(filepath)
            image_data = request.urlopen(i).read()
            file.write(image_data)
        count += 1
def multi_thread(count, func):
    threads = []
    for i in range(0, count):
        threads.append(base_thread(func))
    for j in range(0, count):
        threads[j].start()
    for k in range(0, count):
        threads[k].join()
#CrawlerFunc("http://tieba.baidu.com/p/3764230390")

import paramiko
import paramiko
server_ip = '192.168.1.1'
server_user = 'root'
server_passwd = ''
server_port = 22
ssh = paramiko.SSHClient()
def ssh_connect():
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.load_system_host_keys()
    ssh.connect(server_ip, server_port,server_user, server_passwd)
    return ssh
def client_connect():
    client = paramiko.Transport((server_ip, server_port))
    client.connect(username = server_user, password = server_passwd)
    return client
def ssh_disconnect(client):
    client.close()

def exec_cmd(command, ssh):
    '''
    windows客户端远程执行linux服务器上命令
    '''
    stdin, stdout, stderr = ssh.exec_command(command)
    err = stderr.readline()
    out = stdout.readline()
    print(stdout.read())

def win_to_linux(localpath, remotepath,client):
    '''
    windows向linux服务器上传文件.
    localpath  为本地文件的绝对路径。如：D:  est.py
    remotepath 为服务器端存放上传文件的绝对路径,而不是一个目录。如：/tmp/my_file.txt
    '''

    sftp = paramiko.SFTPClient.from_transport(client)
    sftp.put(localpath,remotepath)
    client.close()

def linux_to_win(localpath, remotepath,client):
    '''
    从linux服务器下载文件到本地
    localpath  为本地文件的绝对路径。如：D:  est.py
    remotepath 为服务器端存放上传文件的绝对路径,而不是一个目录。如：/tmp/my_file.txt
    '''
    sftp = paramiko.SFTPClient.from_transport(client)
    sftp.get(remotepath, localpath)
    client.close()

class AllowAllKeys(paramiko.MissingHostKeyPolicy):
   def missing_host_key(self, client, hostname, key):
       return

def muit_exec_cmd(ssh,cmd):
    '''
    ssh ssh连接
    cmd 多命名
    '''
    ssh.set_missing_host_key_policy(AllowAllKeys())
    channel = ssh.invoke_shell()
    stdin = channel.makefile('wb')
    stdout = channel.makefile('rb')

    stdin.write(cmd)
    print(stdout.read())

    stdout.close()
    stdin.close()

cl = client_connect()
sh = ssh_connect()
muit_exec_cmd(sh,'''
    cd ~
    ls
    sar
    exit
    ''')
win_to_linux("t.txt","/data/t1.txt",cl)
cl.close()
sh.close()