scripts/modeset2.py

116 lines
3.2 KiB
Python
Executable File

#!/usr/bin/env python3
#
# Ferit Yiğit BALABAN <f@fybx.dev>, 2022
#
from datetime import datetime as dt
from subprocess import run
import asyncio
import sys
import os
START_NIGHT = "22.30"
START_DAY = "8.20"
PATH_SCPT_KEYBOARD = "$HOME/scripts/keyboard"
PATH_RESC_VOLUME = "$HOME/.config/navi/volume"
async def connect_keyboard():
command = 'bash ' + os.path.expandvars(PATH_SCRIPT_KEYBOARD)
await open_subprocess(command)
async def set_volume(value: int, save_state = False):
state = get_volume()
if value == -1:
with open(os.path.expandvars(PATH_RESC_VOLUME), 'r') as f:
value = int(f.read())
f.close()
value = 100 if value > 100 else 0 if value < 0 else value
command = f'pactl set-sink-volume @DEFAULT_SINK@ {value}%'
await open_subprocess(command)
if save_state:
with open(os.path.expandvars(PATH_RESC_VOLUME), 'w') as f:
f.write(str(state))
f.close()
async def open_subprocess(cmd: str):
p = await asyncio.create_subprocess_shell(
cmd,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE)
stdout, stderr = await p.communicate()
return p.returncode, stdout, stderr
def get_volume():
r = run(["pactl", "list"], text=True, capture_output=True)
for x in r.stdout.split("Sink #0")[1].split("Base Volume:")[0].split(' '):
if '%' in x:
return int(x.replace('%', ''))
def log(message: str):
with open(os.path.expandvars("$HOME/navi.log"), 'a') as f:
f.write(f"[{dt.now().strftime('%m/%d/%y-%H.%M.%S')}] {message}\n")
f.close()
def get_hour():
if 0 <= dt.now().minute and dt.now().minute <= 9:
return f"{dt.now().hour}.0{dt.now().minute}"
return f"{dt.now().hour}.{dt.now().minute}"
def get_hour_spec(hour_str = None):
if hour_str != None:
return (int(hour_str.split('.')[0]) * 60) + int(hour_str.split('.')[0])
else:
return (dt.now().hour * 60) + dt.now().minute
def get_mode():
low = get_hour_spec(START_DAY)
now = get_hour_spec()
hgh = get_hour_spec(START_NIGHT)
return 0 if low <= now and now < hgh else 1
def main():
sys.argv.remove(sys.argv[0])
sys.argv.reverse()
if len(sys.argv) == 1:
if sys.argv[0] == "--login":
log("modeset2 started with \"--login\"")
asyncio.run(connect_keyboard())
asyncio.run(set_volume(0))
elif sys.argv[0] == "--lock":
log("modeset2 started with \"--lock\"")
print("Lock")
elif sys.argv[0] == "--unlock":
log("modeset2 started with \"--unlock\"")
print("Unlock")
elif sys.argv[0] == "--shutdown":
log("modeset2 started with \"--shutdown\"")
print("Shutdown")
elif len(sys.argv) == 0:
print("modeset2 by fyb")
print(f"local machine time: {get_hour()}")
print(f"current mode is: {get_mode()}")
print(f"current sink volume: {get_volume()}")
print("""Available options:
1. Login
2. Lock
3. Unlock
4. Shutdown""")
else:
print("Issuing more than 1 argument to modeset2 is not supported yet. Aborting...")
exit(1)
exit(0)
if __name__ == '__main__':
main()