python socket mysql_python 中使用with简化MySQL 和 Socket 的连接

数据库 ：

from pymysql import *

class DB(object):

def __init__(self):

self.conn = connect(host='localhost', port=3306, database='python_test_1', user='root', password='mysql', charset='utf8')

self.cos = self.conn.cursor()

# self.num = self.cos.execute()

def __enter__(self):

return self.cos

def __exit__(self, exc_type, exc_val, exc_tb):

self.conn.commit()

self.conn.close()

self.cos.close()

with DB() as db:

num = db.execute("""insert into classes(name) values("python9") """)

socket 的with简化代码 ：

# 导包

from socket import *

# 定义一个套接字的类

class sock(object):

# 把一些固定的语句放入init中

def __init__(self):

self.server_soc = socket(AF_INET, SOCK_STREAM)

self.server_soc.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)

self.server_soc.bind(("",1314))

self.server_soc.listen(1024)

# 将需要返回的数据放入return

def __enter__(self):

self.client_soc, client_addr = self.server_soc.accept()

content = self.client_soc.recv(1024)

return content.decode("utf-8")

# 关闭套接字

def __exit__(self, exc_type, exc_val, exc_tb):

self.server_soc.close()

self.client_soc.close()

while True:

with sock() as soc:

print(soc)