1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
import subprocess
import cv2 as cv
import os
from rich import print
from paths import Paths
def updatei3Theme(config_path: str, img_path: str, colors: list, compliments: list, lock: bool, dmenu: bool):
print('[bold red]Updating i3 color scheme')
data = ''
with open(config_path, 'r') as file:
data = file.readlines()
for i, line in enumerate(data):
# update colors
if "set $bgcolor" in line:
data[i] = 'set $bgcolor ' + colors[0] + '\n'
if "set $in-bgcolor" in line:
data[i] = 'set $in-bgcolor ' + colors[1] + '\n'
if "set $text" in line:
data[i] = 'set $text ' + compliments[0] + '\n'
if "set $indicator" in line:
data[i] = 'set $indicator ' + colors[2] + '\n'
if "set $in-text" in line:
data[i] = 'set $in-text ' + compliments[1] + '\n'
#update background image
if "set $bgimage" in line:
data[i] = 'set $bgimage ' + img_path + '\n'
if "bindsym $mod+d exec --no-startup-id dmenu_run" in line:
if dmenu:
print('[bold red]Updating Dmenu color scheme')
data[i] = f"bindsym $mod+d exec --no-startup-id dmenu_run -nb '{colors[0]}' -sf '{compliments[0]}' -sb '{colors[1]}' -nf '{compliments[1]}'\n"
# update i3lock image, convert to png so it plays nice w i3lock
if lock:
img = cv.imread(img_path)
imgHeight, imgWidth, _ = img.shape
screenWidth, screenHeight = getScreenResolution()
lock_scale = screenWidth / imgWidth
print('[bold red]Creating lock screen')
dim = (int(imgWidth * lock_scale), int(imgHeight * lock_scale))
img = cv.resize(img, dim, interpolation= cv.INTER_AREA)
cv.imwrite('lock.png', img)
os.rename('lock.png', Paths['lockscreen'] + 'lock.png')
with open(config_path, 'w') as file:
file.writelines(data)
print("[bold red]Restarting i3")
os.system("i3 restart")
def getScreenResolution():
output = subprocess.Popen('xrandr | grep "\*" | cut -d" " -f4',shell=True, stdout=subprocess.PIPE).communicate()[0]
resolution = output.split()[0].split(b'x')
width = int(resolution[0].decode('UTF-8'))
height = int(resolution[1].decode('UTF-8'))
return width, height
|
