python打造流媒体服务器_利用python3.5 构建流媒体后台音视频切换的服务端程序...

#!/usr/bin/env python3.5.0#-*- coding:utf8 -*-

importos,sys,socket,hashlib,time,select,threading,configparserimportpymssql

rootdir=os.path.abspath(sys.argv[0])

rootdir=os.path.dirname(rootdir) +"/"cf=configparser.ConfigParser()if os.path.exists(rootdir +'srs_server.conf'):

cf.read(rootdir+'srs_server.conf')else:#如果文件不存在

f = open(rootdir + "/srs_server.conf","w")

f.write("")

f.close()#读取配置文件

cf.read(rootdir +'srs_server.conf')

cf.add_section("srs")

cf.set("srs","config","1")#写回配置文件

cf.write(open(rootdir +'srs_server.conf',"w"))def decrypt(s,key=2):"""解密方法

:param key:

:param s:

:return:"""c= bytearray(str(s).encode("gbk"))

n= len(c) #计算 b 的字节数

if n % 2 !=0 :return ""n= n // 2b=bytearray(n)

j=0for i inrange(0, n):

c1=c[j]

c2= c[j+1]

j= j+2c1= c1 - 65c2= c2 - 65b2= c2*16 +c1

b1= b2^key

b[i]=b1try:return b.decode("gbk")except:return "failed"

#实例化数据库类

classMSSQL:def __init__(self,host,user,pwd,db,port =1433):

self.host=host

self.port=port

self.user=user

self.pwd=pwd

self.db=dbdef __GetConnect(self):try:if notself.db:print("没有设置数据库信息")

self.conn= pymssql.connect(host=self.host,port=self.port,user=self.user,password=self.pwd,database=self.db,charset="utf8")

cur=self.conn.cursor()if notcur:print("连接数据库失败!")else:returncurexceptException as e:print("连接数据库失败,%s"%e)defExecQuery(self,sql):

cur= self.__GetConnect()

cur.execute(sql)

resList=cur.fetchall()#查询完毕后必须关闭连接

self.conn.close()returnresListdefExecNonQuery(self,sql):

cur= self.__GetConnect()

cur.execute(sql)

self.conn.commit()

self.conn.close()#查询数据库相关信息情况#更新SQL数据库相关信息情况

def updatesql(load=0,status=0,item=0):#加载配置文件

cf =configparser.ConfigParser()if os.path.exists(rootdir+"srs_server.conf"):try:

cf.read(rootdir+"srs_server.conf")exceptException as c:print(c)else:print("加载srs_server.conf配置文件失败！")#实例化MSSQL操作类

host = cf.get("HOST","host")

ms= MSSQL(host=cf.get("DB","ip"),user=decrypt(cf.get("DB","username")),pwd=decrypt(cf.get("DB","password")),db=cf.get("DB","db"),port=int(cf.get("DB","port")))try:#更新该服务器所有记录

sql = "update sys_load set item = '%s',sysload='%s',status='%s' where servername ='%s'"%(item, load, status, host)

ms.ExecNonQuery(sql)except:pass

defselectdata(conn):

conf= cf.get("srs","config")if conf == "1":

conn.send(bytes("当前为音频模式！","utf-8"))elif conf == "2":

conn.send(bytes("当前为视频模式！","utf-8"))defupdatesound(conn):

with open(rootdir+'srs.conf',"r") as f:

data=f.readlines()for i,line inenumerate(data):#去除制表符，换行符

line = line.replace('\t','').replace('\n','').replace(' ','')if line =="#allowpublish220.168.91.254;":

data[i]="\tallow\t\tpublish\t\t220.168.91.254;\t\n"

elif line == "#forward10.27.68.150;":

data[i]= "\tforward\t\t10.27.68.150;\t\t \n"

elif line == "transcodelive/livestream{":

f.seek(i+1)

data[i+1] = "\tenabled\t\ton;\n"with open(rootdir+'srs.conf',"w") as xf:

xf.writelines(data)

xf.close()

status= os.system("service srs restart")if status ==0:

cf.read(rootdir+'srs_server.conf')

cf.set("srs","config","1")

cf.write(open(rootdir+'srs_server.conf',"w"))

updatesql(item=0,load=1,status=0)

conn.send(bytes("当前已成功切换为音频模式","utf-8"))else:

conn.send(bytes("切换为音频模式失败，请联系IT部！","utf-8"))defupdatevideo(conn):

with open(rootdir+'srs.conf',"r") as f:

data=f.readlines()for i,line inenumerate(data):#去除制表符，换行符

line = line.replace('\t','').replace('\n','').replace(' ','')if line =="allowpublish220.168.91.254;":

data[i]="\t#allow\t\tpublish\t\t220.168.91.254;\t\n"

elif line == "forward10.27.68.150;":

data[i]= "\t#forward\t10.27.68.150;\n"

elif line == "transcodelive/livestream{":#下一行

f.seek(i+1)

data[i+1] = "\tenabled\t\toff;\n"with open(rootdir+'srs.conf',"w") as xf:

xf.writelines(data)

xf.close()

status= os.system("service srs restart")if status ==0:

cf.read(rootdir+'srs_server.conf')

cf.set("srs","config","2")

cf.write(open(rootdir+'srs_server.conf',"w"))

updatesql(item=1,load=3,status=1)

conn.send(bytes("当前已成功切换为视频模式","utf-8"))else:

conn.send(bytes("切换视频模式失败，请联系IT部！","utf-8"))defcrash_recovery(conn):

os.system("service srs stop")#强制还原配置文件

os.system("yes|cp -fr srs_bak.conf srs.conf")

status= os.system("service srs start")if status ==0:

cf.read(rootdir+'srs_server.conf')

cf.set("srs","config","1")

cf.write(open(rootdir+'srs_server.conf',"w"))

updatesql(item=0,load=1,status=0)

conn.send(bytes("修复音频模式成功","utf-8"))else:

conn.send(bytes("修复音频模式失败，请与IT部联系！","utf-8"))#业务执行函数

defoperation(conn):

content= """请输入数字选项进行操作：

1、查询当前直播频道

2、切换频道为音频直播

3、切换频道为视频直播

4、故障修复(默认修复为音频模式)

***********************"""conn.send(bytes(content,"utf-8"))whileTrue:#接收数据

data = conn.recv(1000)

test=data.decode()if int(test) == 1:

selectdata(conn)elif int(test) ==2:

updatesound(conn)elif int(test) ==3:

updatevideo(conn)elif int(test) ==4:

crash_recovery(conn)#SOCKET主进程模块

defprocess(conn,addr):try:

i=0#认证失败允许重试3次

while i < 3:

flage=False#接收客户端连接请求信息

info = conn.recv(1000)#实例化加密函数

hash =hashlib.sha512()

hash.update("a123456789B".encode("utf-8")) #KEY＝a123456789B

hash_pwd =hash.hexdigest()if info.decode() ==hash_pwd:

hash_pwd= ""info= ""

if addr[0] in ["192.168.1.252","192.168.1.190","127.0.0.1"]:

flage=True#接收用户及密码信息

whileflage:

operation(conn)else:#登陆失败，发送给客户端重新验证

i += 1conn.send(bytes("error","utf8"))if i > 2:#主动关闭连接

conn.close()

time.sleep(25)exceptException as e:

conn.close()

time.sleep(15)#SOCKET服务模块

defsock_server():'''启动服务器端，开启线程监听

:return:'''server=socket.socket()

server_ip="localhost"server_port= 1888server.bind((server_ip,server_port))

server.listen(10)whileTrue:

r,w,e= select.select([server,], [], [], 1)for i,server inenumerate(r):

time.sleep(1)

conn,addr=server.accept()#创建线程

t = threading.Thread(target=process, args=(conn, addr))#启动线程

t.start()if __name__ =="__main__":

sock_server()