9 changed files with 368 additions and 157 deletions
Split View
Diff Options
-
+163 -0,
-
BIN.gradle/6.8/fileChanges/last-build.bin
-
BIN.gradle/6.8/fileHashes/fileHashes.lock
-
+0 -0.gradle/6.8/gc.properties
-
BIN.gradle/checksums/checksums.lock
-
+0 -0.gradle/configuration-cache/gc.properties
-
+0 -0.gradle/vcs-1/gc.properties
-
+73 -68get_meetings.py
-
+132 -89main.py
+ 163
- 0
,
View File
| @ -0,0 +1,163 @@ | |||
| import get_meetings | |||
| import subprocess | |||
| import datetime | |||
| import os | |||
| import json | |||
| import time | |||
| import sys | |||
| 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"] | |||
| 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") | |||
| while True: | |||
| 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") | |||
| if AUTO_LOGIN: | |||
| for m in meetings: | |||
| print("{}:{} {}".format(m["time"],m["class"],m["http_url"])) | |||
| 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) | |||