pulizia codice mumblebot
This commit is contained in:
parent
e27aeaff68
commit
579b1a07cf
1 changed files with 112 additions and 83 deletions
|
|
@ -1,107 +1,136 @@
|
||||||
import pymumble_py3 as pymumble
|
|
||||||
from pymumble_py3.constants import *
|
|
||||||
import subprocess as sp
|
|
||||||
import time
|
|
||||||
import sys
|
|
||||||
import os
|
|
||||||
import fcntl
|
|
||||||
import audioop
|
|
||||||
import argparse
|
import argparse
|
||||||
|
import audioop
|
||||||
|
import logging
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
from functools import partial
|
||||||
|
|
||||||
def message_received(message):
|
import pymumble_py3 as pymumble
|
||||||
|
from pymumble_py3.constants import PYMUMBLE_CLBK_TEXTMESSAGERECEIVED
|
||||||
|
|
||||||
|
is_streaming = False
|
||||||
|
silence_time = 0
|
||||||
|
|
||||||
|
|
||||||
|
def message_received(mumble, message):
|
||||||
global is_streaming
|
global is_streaming
|
||||||
global silence_time
|
global silence_time
|
||||||
command=message.message
|
command = message.message
|
||||||
if command == "/start":
|
if command == "/start":
|
||||||
is_streaming = True
|
is_streaming = True
|
||||||
silence_time = 0
|
silence_time = 0
|
||||||
mumble.my_channel().send_text_message("Diretta iniziata")
|
mumble.my_channel().send_text_message("Diretta iniziata")
|
||||||
|
logging.info("Diretta iniziata")
|
||||||
mumble.users.myself.recording()
|
mumble.users.myself.recording()
|
||||||
elif command == "/stop":
|
elif command == "/stop":
|
||||||
is_streaming = False
|
is_streaming = False
|
||||||
mumble.my_channel().send_text_message("Diretta terminata")
|
mumble.my_channel().send_text_message("Diretta terminata")
|
||||||
|
logging.info("Diretta terminata")
|
||||||
mumble.users.myself.unrecording()
|
mumble.users.myself.unrecording()
|
||||||
|
|
||||||
parser = argparse.ArgumentParser(description='Regia pienamente automatizzata')
|
|
||||||
parser.add_argument('--channel', help='Set channel', default="")
|
|
||||||
parser.add_argument('--name', help='Set bot nickname', default="RadioRobbot")
|
|
||||||
parser.add_argument('--server', help='Set server', default="mumble.esiliati.org")
|
|
||||||
parser.add_argument('--port', help='Set port', type=int, default=64738)
|
|
||||||
parser.add_argument('--stream', action='store_true', help='Ignore commands in chat and stream everything')
|
|
||||||
parser.add_argument('--auto-suspend-stream', action='store_true', help='Ignore commands in chat and stream everything')
|
|
||||||
parser.add_argument('--max-silence-time', type=int, help='max silence time in seconds', default=30)
|
|
||||||
|
|
||||||
sys.argv.pop(0)
|
|
||||||
args = parser.parse_args(sys.argv)
|
|
||||||
|
|
||||||
|
|
||||||
pwd = "" # password
|
|
||||||
server = args.server
|
|
||||||
nick = args.name
|
|
||||||
channel = args.channel
|
|
||||||
port = args.port
|
|
||||||
is_streaming=False
|
|
||||||
stream_always= args.stream
|
|
||||||
auto_suspend_stream = args.auto_suspend_stream
|
|
||||||
silence_limit = 30
|
|
||||||
|
|
||||||
|
|
||||||
|
def get_parser():
|
||||||
|
parser = argparse.ArgumentParser(description="Regia pienamente automatizzata")
|
||||||
|
parser.add_argument("--channel", help="Set channel", default="")
|
||||||
|
parser.add_argument("--name", help="Set bot nickname", default="RadioRobbot")
|
||||||
|
parser.add_argument("--server", help="Set server", default="mumble.esiliati.org")
|
||||||
|
parser.add_argument("--port", help="Set port", type=int, default=64738)
|
||||||
|
parser.add_argument(
|
||||||
|
"--stream",
|
||||||
|
action="store_true",
|
||||||
|
help="Ignore commands in chat and stream everything",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--auto-suspend-stream",
|
||||||
|
action="store_true",
|
||||||
|
help="Ignore commands in chat and stream everything",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--max-silence-time", type=int, help="max silence time in seconds", default=30
|
||||||
|
)
|
||||||
|
return parser
|
||||||
|
|
||||||
# Spin up a client and connect to mumble server
|
|
||||||
mumble = pymumble.Mumble(server, nick, password=pwd, port=port)
|
|
||||||
|
|
||||||
mumble.callbacks.set_callback(PYMUMBLE_CLBK_TEXTMESSAGERECEIVED, message_received)
|
def main():
|
||||||
mumble.set_receive_sound(1) # Enable receiving sound from mumble server
|
global is_streaming
|
||||||
mumble.start()
|
global silence_time
|
||||||
mumble.is_ready() # Wait for client is ready
|
args = get_parser().parse_args()
|
||||||
mumble.channels.find_by_name(channel).move_in()
|
logging.basicConfig(level=logging.DEBUG)
|
||||||
mumble.users.myself.mute()
|
|
||||||
|
|
||||||
if is_streaming:
|
pwd = "" # password
|
||||||
mumble.users.myself.recording()
|
server = args.server
|
||||||
|
nick = args.name
|
||||||
|
channel = args.channel
|
||||||
|
port = args.port
|
||||||
|
is_streaming = False
|
||||||
|
stream_always = args.stream
|
||||||
|
|
||||||
BUFFER = 0.1
|
# Spin up a client and connect to mumble server
|
||||||
BITRATE = 48000
|
mumble = pymumble.Mumble(server, nick, password=pwd, port=port)
|
||||||
RESOLUTION = 10 # in ms
|
|
||||||
FLOAT_RESOLUTION = float(RESOLUTION) / 1000
|
|
||||||
MONO_CHUNK_SIZE = BITRATE * 2 * RESOLUTION / 1000
|
|
||||||
STEREO_CHUNK_SIZE = MONO_CHUNK_SIZE * 2
|
|
||||||
silent = b"\x00" * int(STEREO_CHUNK_SIZE)
|
|
||||||
cursor_time = None
|
|
||||||
cursor_time = time.time() - BUFFER
|
|
||||||
silence_time = 0
|
|
||||||
silence_limit_ms = silence_limit * 1000
|
|
||||||
|
|
||||||
while mumble.is_alive():
|
mumble.callbacks.set_callback(
|
||||||
if cursor_time < time.time() - BUFFER:
|
PYMUMBLE_CLBK_TEXTMESSAGERECEIVED, partial(message_received, mumble)
|
||||||
base_sound = None
|
)
|
||||||
|
mumble.set_receive_sound(1) # Enable receiving sound from mumble server
|
||||||
|
mumble.start()
|
||||||
|
mumble.is_ready() # Wait for client is ready
|
||||||
|
mumble.channels.find_by_name(channel).move_in()
|
||||||
|
mumble.users.myself.mute()
|
||||||
|
|
||||||
try:
|
if is_streaming:
|
||||||
for user in mumble.users.values(): # check the audio queue of each user
|
mumble.users.myself.recording()
|
||||||
if user.sound.is_sound():
|
|
||||||
# available sound is to be treated now and not later
|
|
||||||
sound = user.sound.get_sound(FLOAT_RESOLUTION)
|
|
||||||
stereo_pcm = audioop.tostereo(sound.pcm, 2, 1, 1)
|
|
||||||
if base_sound == None:
|
|
||||||
base_sound = stereo_pcm
|
|
||||||
else:
|
|
||||||
base_sound = audioop.add(base_sound, stereo_pcm, 2)
|
|
||||||
except RuntimeError:
|
|
||||||
print("ignored exception in stderr...", file=sys.stderr)
|
|
||||||
|
|
||||||
if is_streaming or stream_always:
|
BUFFER = 0.1
|
||||||
if base_sound:
|
BITRATE = 48000
|
||||||
silence_time = 0
|
RESOLUTION = 10 # in ms
|
||||||
sys.stdout.buffer.write(base_sound)
|
FLOAT_RESOLUTION = float(RESOLUTION) / 1000
|
||||||
else:
|
MONO_CHUNK_SIZE = BITRATE * 2 * RESOLUTION / 1000
|
||||||
silence_time += RESOLUTION
|
STEREO_CHUNK_SIZE = MONO_CHUNK_SIZE * 2
|
||||||
sys.stdout.buffer.write(silent)
|
silent = b"\x00" * int(STEREO_CHUNK_SIZE)
|
||||||
|
cursor_time = time.time() - BUFFER
|
||||||
|
|
||||||
if auto_suspend_stream and (silence_time >= silence_limit_ms) and is_streaming:
|
while mumble.is_alive():
|
||||||
is_streaming = False
|
if cursor_time < time.time() - BUFFER:
|
||||||
mumble.my_channel().send_text_message("Diretta terminata in automatico dopo "+str(silence_limit)+" secondi circa di silenzio")
|
base_sound = None
|
||||||
mumble.users.myself.unrecording()
|
|
||||||
|
|
||||||
cursor_time += FLOAT_RESOLUTION
|
try:
|
||||||
else:
|
for user in mumble.users.values(): # check the audio queue of each user
|
||||||
time.sleep(FLOAT_RESOLUTION)
|
if user.sound.is_sound():
|
||||||
|
# available sound is to be treated now and not later
|
||||||
|
sound = user.sound.get_sound(FLOAT_RESOLUTION)
|
||||||
|
stereo_pcm = audioop.tostereo(sound.pcm, 2, 1, 1)
|
||||||
|
if base_sound is None:
|
||||||
|
base_sound = stereo_pcm
|
||||||
|
else:
|
||||||
|
base_sound = audioop.add(base_sound, stereo_pcm, 2)
|
||||||
|
except RuntimeError:
|
||||||
|
print("ignored exception in stderr...", file=sys.stderr)
|
||||||
|
|
||||||
|
if is_streaming or stream_always:
|
||||||
|
if base_sound:
|
||||||
|
silence_time = 0
|
||||||
|
sys.stdout.buffer.write(base_sound)
|
||||||
|
else:
|
||||||
|
silence_time += RESOLUTION
|
||||||
|
sys.stdout.buffer.write(silent)
|
||||||
|
|
||||||
|
if (
|
||||||
|
args.auto_suspend_stream
|
||||||
|
and (silence_time >= args.max_silence_time * 1000)
|
||||||
|
and is_streaming
|
||||||
|
):
|
||||||
|
is_streaming = False
|
||||||
|
logging.info("max-silence-time reached")
|
||||||
|
mumble.my_channel().send_text_message(
|
||||||
|
"Diretta terminata in automatico dopo %d secondi circa di silenzio"
|
||||||
|
% args.max_silence_time
|
||||||
|
)
|
||||||
|
mumble.users.myself.unrecording()
|
||||||
|
|
||||||
|
cursor_time += FLOAT_RESOLUTION
|
||||||
|
else:
|
||||||
|
time.sleep(FLOAT_RESOLUTION)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue