You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
179 lines
7.0 KiB
179 lines
7.0 KiB
import get_meetings
|
|
import subprocess
|
|
import datetime
|
|
import os
|
|
import json
|
|
import time
|
|
import sys
|
|
import discord
|
|
from smb.SMBConnection import SMBConnection
|
|
|
|
RECORD_PATH = "recordings"
|
|
LISTENING_VIDEO_PATH = "listening.mkv"
|
|
VIRTUAL_CAMERA_PATH = "/dev/video2"
|
|
PULSEAUDIO_INDEX = 5
|
|
X_OFFSET = sys.argv[1]
|
|
PULSEAUDIO_INDEX = sys.argv[2]
|
|
AUTO_LOGIN = sys.argv[3] == 1
|
|
|
|
RECORD_TEMPLATE = ["ffmpeg", "-nostdin", "-video_size", "1920x1080", "-f", "x11grab", "-i", ":0.0+{},0".format(X_OFFSET), "-f", "pulse", "-ac", "2", "-i", str(PULSEAUDIO_INDEX), "-vcodec", "libx264", "-crf", "0", "-preset", "ultrafast", "-acodec", "pcm_s16le"]
|
|
CAMERA_TEMPLATE = ["ffmpeg", "-re", "-i", LISTENING_VIDEO_PATH, "-map", "0:v", "-f", "v4l2", VIRTUAL_CAMERA_PATH]
|
|
COMPRESSION_TEMPLATE = ["ffmpeg", "-i", None, "-vcodec", "libx265", "-crf", "28", None]
|
|
|
|
|
|
if not os.path.exists(VIRTUAL_CAMERA_PATH):
|
|
os.system("sudo modprobe -r v4l2loopback")
|
|
os.system("sudo modprobe v4l2loopback")
|
|
|
|
if not os.path.exists(RECORD_PATH):
|
|
os.mkdir(RECORD_PATH)
|
|
|
|
with open("config.json", "r") as f:
|
|
config = json.loads(f.read())
|
|
creds = config["creds"]
|
|
backup_creds = config["backup"]
|
|
DISCORD = config["discord"]
|
|
|
|
next_day = datetime.datetime.today()
|
|
|
|
def backup(directory):
|
|
conn = SMBConnection(backup_creds["user"], backup_creds["password"],
|
|
backup_creds["server_name"], backup_creds["server_name"], use_ntlm_v2 = True)
|
|
conn.connect(backup_creds["ip"])
|
|
path = "/"
|
|
files = conn.listPath(backup_creds["share"], path)
|
|
for i in backup_creds["path"].split("/"):
|
|
exists = False
|
|
path += i + "/"
|
|
for f in files:
|
|
if f.filename == i:
|
|
if f.isDirectory:
|
|
exists = True
|
|
break
|
|
if not exists:
|
|
try:
|
|
conn.createDirectory(backup_creds["share"],path)
|
|
except Exception as e:
|
|
print(e)
|
|
return 1
|
|
files = conn.listPath(backup_creds["share"], path)
|
|
|
|
folder = conn.getAttributes(backup_creds["share"], backup_creds["path"])
|
|
if folder.isReadOnly:
|
|
print("BACKUP PATH IS READONLY")
|
|
return 1
|
|
|
|
local_path = os.path.join(RECORD_PATH, directory)
|
|
remote_path = os.path.join(backup_creds["path"], directory)
|
|
files = conn.listPath(backup_creds["share"], backup_creds["path"])
|
|
for f in files:
|
|
if f.filename == directory:
|
|
print("ALREADY BACKED UP")
|
|
return
|
|
conn.createDirectory(backup_creds["share"], remote_path)
|
|
print("STARTING BACKUP")
|
|
for i in os.listdir(local_path):
|
|
with open(os.path.join(local_path, i), "rb") as f:
|
|
print(i)
|
|
conn.storeFile(backup_creds["share"],
|
|
os.path.join(remote_path, i),
|
|
f)
|
|
conn.close()
|
|
print("DONE BACKING UP")
|
|
|
|
if(not AUTO_LOGIN):
|
|
client = discord.Client()
|
|
|
|
@client.event
|
|
async def on_ready():
|
|
print(f'{client.user} has connected to Discord!')
|
|
meetings = get_meetings.get_meetings(creds["TC"], creds["passwd"])
|
|
msg = ""
|
|
for m in meetings:
|
|
msg += "{}: **{}** <{}>\n".format(m["time"],m["class"],m["http_url"])
|
|
await client.get_guild(int(DISCORD["guild"])).get_channel(int(DISCORD["channel"])).send(msg)
|
|
await client.close()
|
|
|
|
client.run(DISCORD["token"])
|
|
|
|
while True:
|
|
if not AUTO_LOGIN:
|
|
break
|
|
now = datetime.datetime.today()
|
|
while now < next_day:
|
|
time.sleep(7200)
|
|
now = datetime.datetime.today()
|
|
print("NEW DAY")
|
|
print("GENERATING MEETING LIST")
|
|
meetings = get_meetings.get_meetings(creds["TC"], creds["passwd"])
|
|
print("DONE\n")
|
|
|
|
day = datetime.datetime.today().strftime("%d_%m_%Y")
|
|
next_day = datetime.datetime.strptime("{} 00:00:00".format(day), '%d_%m_%Y %H:%M:%S') + datetime.timedelta(days = 1)
|
|
if not os.path.exists(os.path.join(RECORD_PATH, day)):
|
|
os.mkdir(os.path.join(RECORD_PATH, day))
|
|
video_recording = None
|
|
camera_mirroring = None
|
|
k = None
|
|
for k,i in enumerate(meetings):
|
|
now = datetime.datetime.today()
|
|
meeting_time = datetime.datetime.strptime("{} {}:00".format(day, i["time"]), '%d_%m_%Y %H:%M:%S')
|
|
end_time = datetime.datetime.strptime("{} {}:00".format(day, i["time"]), '%d_%m_%Y %H:%M:%S')+ datetime.timedelta(minutes = 30)
|
|
if meeting_time < now:
|
|
continue
|
|
if camera_mirroring:
|
|
while now < end_time:
|
|
time.sleep(30)
|
|
now = datetime.datetime.today()
|
|
print("CLASS FINISHED, TERMINATING PREVIOUS RECORDINGS\n")
|
|
if video_recording:
|
|
video_recording.terminate()
|
|
video_recording = None
|
|
camera_mirroring.terminate()
|
|
print(i["time"])
|
|
print(i["class"])
|
|
while now < meeting_time:
|
|
time.sleep(30)
|
|
now = datetime.datetime.today()
|
|
print("STARTING")
|
|
kill_proc = subprocess.Popen(["killall","-9","zoom"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
kill_proc.wait() # Kill all zoom instances before entering a new meeting
|
|
subprocess.Popen(["xdg-open", i["meeting_url"]], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
cmd = RECORD_TEMPLATE + [os.path.join(RECORD_PATH, day, "{}.{}.mkv".format(k, i["class"]))]
|
|
if os.path.exists(os.path.join(RECORD_PATH, day, "{}.{}.mkv".format(k, i["class"]))):
|
|
os.remove(os.path.join(RECORD_PATH, day, "{}.{}.mkv".format(k, i["class"])))
|
|
blacklisted = False
|
|
for j in config["blacklisted"]:
|
|
if j in i["class"]:
|
|
blacklisted = True
|
|
if not blacklisted:
|
|
video_recording = subprocess.Popen(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
camera_mirroring = subprocess.Popen(CAMERA_TEMPLATE, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
if k:
|
|
end_time = datetime.datetime.strptime("{} {}:00".format(day, meetings[k]["time"]), '%d_%m_%Y %H:%M:%S')+ datetime.timedelta(minutes = 30)
|
|
while now < end_time:
|
|
time.sleep(30)
|
|
now = datetime.datetime.today()
|
|
print("CLASS FINISHED, TERMINATING PREVIOUS RECORDINGS\n")
|
|
if video_recording:
|
|
video_recording.terminate()
|
|
if camera_mirroring:
|
|
camera_mirroring.terminate()
|
|
print("\nALL CLASSES RECORDED\n")
|
|
todays_recordings_path = os.path.join(RECORD_PATH, day)
|
|
recordings = os.listdir(todays_recordings_path)
|
|
for r in recordings:
|
|
if r[-3:] != "mkv":
|
|
continue
|
|
COMPRESSION_TEMPLATE[2] = os.path.join(todays_recordings_path, r)
|
|
COMPRESSION_TEMPLATE[-1] = os.path.join(todays_recordings_path, r[:-3] + ".mp4")
|
|
print("COMPRESSING " + r)
|
|
compression_process = subprocess.Popen(COMPRESSION_TEMPLATE, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
exit_code = compression_process.wait()
|
|
print("FFMPEG EXITED WITH STATUS CODE {}".format(exit_code))
|
|
if exit_code == 0:
|
|
os.remove(COMPRESSION_TEMPLATE[2])
|
|
|
|
backup(day)
|
|
|
|
|