Python之路-python(面向对象进阶（模块的动态导入、断言、Socket Server）)

模块的动态导入

断言

Socket Server

 

　　一、模块的动态导入

class C(object):
    def __init__(self):
        self.name = "zhangsan"

#动态导入
import  importlib
test = importlib.import_module("lib.test")
print(test.C().name)

 

　　二、断言

　　

　　assert断言是一句必须等价于布尔真的判定！

 

　　1 不等于 0 就会有AssertionError异常

 

　　1 等于 0 就没有异常

 

　　如果断言成功（如果为真）那么不执行任何操作！

 

　　如果断言不成功，那么会触发AssertionError

try:
    assert True == 0
except AssertionError as e:
    print("断言不成立")
else:
    print("断言成立")
>>>断言不成立

try:
    assert True == 1
except AssertionError as e:
    print("断言不成立")
else:
    print("断言成立")
>>>断言成立

 

　　三、Socket Server

 

Socket Families(地址簇)

socket.AF_UNIX unix本机进程间通信 

socket.AF_INET　IPV4　

socket.AF_INET6  IPV6

Socket Types

socket.SOCK_STREAM  #for tcp

socket.SOCK_DGRAM   #for udp 

socket.SOCK_RAW     #原始套接字，普通的套接字无法处理ICMP、IGMP等网络报文，而SOCK_RAW可以；其次，SOCK_RAW也可以处理特殊的IPv4报文；此外，利用原始套接字，可以通过IP_HDRINCL套接字选项由用户构造IP头。

socket.SOCK_RDM  #是一种可靠的UDP形式，即保证交付数据报但不保证顺序。SOCK_RAM用来提供对原始协议的低级访问，在需要执行某些特殊操作时使用，如发送ICMP报文。SOCK_RAM通常仅限于高级用户或管理员运行的程序使用。

socket.SOCK_SEQPACKET #废弃了

Socket 方法

sk.bind(address)

　　s.bind(address) 将套接字绑定到地址。address地址的格式取决于地址族。在AF_INET下，以元组（host,port）的形式表示地址。

sk.listen(backlog)

　　开始监听传入连接。backlog指定在拒绝连接之前，可以挂起的最大连接数量。

      backlog等于5，表示内核已经接到了连接请求，但服务器还没有调用accept进行处理的连接个数最大为5
      这个值不能无限大，因为要在内核中维护连接队列

sk.setblocking(bool)

　　是否阻塞（默认True），如果设置False，那么accept和recv时一旦无数据，则报错。

sk.accept()

　　接受连接并返回（conn,address）,其中conn是新的套接字对象，可以用来接收和发送数据。address是连接客户端的地址。

　　接收TCP 客户的连接（阻塞式）等待连接的到来

sk.connect(address)

　　连接到address处的套接字。一般，address的格式为元组（hostname,port）,如果连接出错，返回socket.error错误。

sk.connect_ex(address)

　　同上，只不过会有返回值，连接成功时返回 0 ，连接失败时候返回编码，例如：10061

sk.close()

　　关闭套接字

sk.recv(bufsize[,flag])

　　接受套接字的数据。数据以字符串形式返回，bufsize指定最多可以接收的数量。flag提供有关消息的其他信息，通常可以忽略。

sk.recvfrom(bufsize[.flag])

　　与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。

sk.send(string[,flag])

　　将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。即：可能未将指定内容全部发送。

sk.sendall(string[,flag])

　　将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。

      内部通过递归调用send，将所有内容发送出去。

sk.sendto(string[,flag],address)

　　将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数。该函数主要用于UDP协议。

sk.settimeout(timeout)

　　设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如 client 连接最多等待5s ）

sk.getpeername()

　　返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。

sk.getsockname()

　　返回套接字自己的地址。通常是一个元组(ipaddr,port)

sk.fileno()

　　套接字的文件描述符

socket.sendfile(file, offset=0, count=None)

     发送文件 。

简单SSH代码（处理了粘包问题，但不能返回top命令执行的结果）

Socket_Server

import socket,os

#生成句柄。
Server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
#绑定ip地址和端口。
Server.bind(("localhost",9999))
#开始监听。
Server.listen()
while True:
    #阻塞状态，无限循环等待客户端连接。
    print("阻塞状态，等待指令！！！")
    conn,addr = Server.accept()
    while True:
        # 无限循环，等待每个对象的命令，并执行和其他操作。
        data = conn.recv(1024)
        #如果命令为空，就返回上一级循环
        if not data:break
        #把接收到的命令进行转码
        data = data.decode("utf-8")
        #打印接收到的命令
        print("执行命令---->>>>\n:",data)
        #使用os模块执行客户端发来的命令
        result = os.popen(data).read()
        #打印命令执行结果
        # print(result)
        #如果执行命令长度为0，就说明命令没有执行成功
        if len(result) == 0:print("命令执行失败")
        #先发送执行结果的长度，注意：linux和windows编码不同，所以需要以下写法。
        #result原始数据里面可能有中文和英文的区别，所以我们一定要先转成utf-8以后计算长度，
        #这个长度说白了就是以最长计算（gbk和utf-8，中文最长），避免少收取。最后再把这个长度以str字符串的方式发给客户端，但
        #要提前设置成编码为utf-8
        conn.send(str(len(result.encode())).encode("utf-8"))
        #下面通过再和客户端交互一次的方式，让前后两个send在内存中不存在粘包问题，你嫌麻烦不限low可以用time.sleep(0.5)
        check_ack = conn.recv(1024)
        print(check_ack.decode())
        #一次性将命令结果发送到客户端
        conn.send(result.encode("utf-8"))


###注意最后连续两次send会造成粘包问题，linux上最明显。

 

Socket_Client

import socket

#生成客户端句柄
Client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
#连接服务器
Client.connect(("localhost",9999))
while True:
    #用户输入要执行的命令(此处还不支持命令之间有空格)
    rece_data = input("请输入命令：").strip()
    #如果用户未输入命令，就继续循环让其输入命令
    if len(rece_data) == 0:continue
    #否则就将命令发送至服务器，注意编码
    Client.send(rece_data.encode("utf-8"))
    #接收，服务器发来的命令长度
    cmd_size = Client.recv(1024)
    #如果命令返回的结果长度等于1，就说明没有此命令(会返回一个b'0',但len显示的是1)
    if len(cmd_size.decode()) == 1:print("无此命令")
    #下面一行是处理粘包问题
    Client.send("可以继续发送了，不会粘包的".encode("utf-8"))
    #设置默认接收长度为0
    receive_size = 0
    #设置默认初始数据为空
    receive_data = b''
    #如果服务器返回命令执行结果的数值大于0的话，就进入循环。
    while int(cmd_size.decode()) > receive_size:
        #先接收1024长度的数据
        data = Client.recv(1024)
        #设置，receive_size = receive_size + len(data)
        receive_size += len(data)
        #将每次收到的数据都加到一起，最后打印的时候再decode，因为上面用的是b''
        receive_data += data
    else:
        #打印最后收到的数据长度
        print("receive_size:",receive_size)
        #打印所有收到的数据，ignore是忽略gbk和utf-8转码中的报错，默认是strict
        print(receive_data.decode("utf-8"))
Client.close()

 

服务器端

import  socketserver

class MyTCPHandler(socketserver.BaseRequestHandler):
    def handle(self):
        while True:
            try:
                self.data = self.request.recv(1024).strip()
                print("{} wrote:".format(self.client_address[0]))
                print(self.data)
                self.request.send(self.data.upper())
            except ConnectionResetError as e:
                print("error",e)
                break
if __name__ == '__main__':
    HOST,PORT = "localhost",9999
    server = socketserver.ThreadingTCPServer((HOST,PORT),MyTCPHandler)
    server.serve_forever()

#客户端
import socket
client = socket.socket() #声明socket类型，同时生成socket连接对象
#client.connect(('192.168.16.200',9999))
client.connect(('localhost',9999))
while True:
    msg = input(">>:").strip()
    if len(msg) == 0:continue
    client.send(msg.encode("utf-8"))
    data = client.recv(10240)
    print("recv:",data.decode())
client.close()

#ftp服务端
import os,hashlib
import  socket
server = socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
    conn,addr = server.accept()
    print("new conn:",addr)
    while True:
        print("等待新指令")
        data = conn.recv(1024)
        if not data:
            print("客户端已经断开")
            break
        cmd,filename = data.decode().split()
        print(filename)
        if os.path.isfile(filename):
           f = open(filename,"rb")
           m = hashlib.md5()
           file_size = os.stat(filename).st_size
           conn.send(str(file_size).encode())
           conn.recv(1024)
           for line in f:
               m.update(line)
               conn.send(line)
           print("file md5",m.hexdigest())
           f.close()
           conn.send(m.hexdigest().encode())
           print("send done")
server.close()

#ftp客户端
import  socket,hashlib
client = socket.socket()
client.connect(("localhost",9999))
while True:
    cmd = input(">>：").strip()
    if len(cmd) == 0:continue
    if cmd.startswith("get"):
        client.send(cmd.encode())
        server_response = client.recv(1024)
        print("server response:",server_response)
        client.send(b"ready to recv file")
        file_total_size = int(server_response.decode())
        received_size = 0
        filename = cmd.split()[1]
        f = open(filename+".new","wb")
        m = hashlib.md5()
        while received_size < file_total_size:
            if file_total_size - received_size > 1024:#要收不止一次
                size = 1024
            else:##最后一次了，剩多少收多少
                size = file_total_size - received_size
                print("last receive:",size)
            data  = client.recv(size)
            received_size += len(data)
            m.update(data)
            f.write(data)
        else:
            new_file_md5 = m.hexdigest()
            print("file recv done",received_size,file_total_size)
            f.close()
        server_file_md5 = client.recv(1024)
        print("server file md5",server_file_md5)
        print("client file md5",new_file_md5)
client.close()