sched.py

bashrc

@@ -7,6 +7,8 @@
 PS1='\[\e[0;1;91m\][\[\e[0;1;38;5;87m\]\u\[\e[0;0;38;5;123m\]@\[\e[0;0;38;5;159m\]\h \[\e[0;38;5;247m\]\w\[\e[0;1;91m\]]\n \[\e[0;38;5;247m\]$ \[\e[0m\]'
 PROMPT_COMMAND="echo"                           # Line jump after every command
 
+[ -e "/usr/bin/afetch" ] && afetch || fetch min
+
 # General configs
 PATH="$HOME/.config/scripts${PATH:+:${PATH}}"   # Enables custom scripts dir
 bind "set completion-ignore-case on"            # Case insensitive TAB complete
@@ -117,5 +119,3 @@ type systemctl 2>/dev/null 1>&2 && if systemctl -q is-active graphical.target &&
     amixer
     exec startx
 fi
-
-[ -e "/usr/bin/afetch" ] && afetch || fetch min

scripts/sched.py

@@ -0,0 +1,195 @@
+#!/usr/bin/env python
+help="Usage: sched.py [OPTION] [FILE]\nOptions: new, print, edit."
+import sys  # arguments (if given file, just output it with formatting)
+import os   # clear console for easier readability
+
+# schedule layout & formatting
+weekdays = [
+    "sun",
+    "mon",
+    "tue",
+    "wed",
+    "thu",
+    "fri",
+    "sat",
+    ]
+hours = [
+    "05:00",
+    "06:00",
+    "07:00",
+    "08:00",
+    "09:00",
+    "10:00",
+    "11:00",
+    "12:00",
+    "13:00",
+    "14:00",
+    "15:00",
+    "16:00",
+    "17:00",
+    "18:00",
+    "19:00",
+    "20:00",
+    ]
+colors = [
+    '\033[0m',  # 0 - normal
+    '\033[90m', # 1 - grey
+    '\033[91m', # 2 - red
+    '\033[92m', # 3 - green
+    '\033[93m', # 4 - yellow
+    '\033[94m', # 5 - purple
+    '\033[95m', # 6 - magenta
+    '\033[96m', # 7 - cyan
+    ]
+styles = [
+    '\033[0m',  # 0 - normal
+    '\033[1m',  # 1 - bold
+    '\033[4m',  # 2 - underline
+    ]
+header_align = "center" # center, left, right
+entry_align  = "left"   # center, left, right
+cell_width   = 13
+header_color = 1
+header_style = 0
+
+# generate from previous values (utf8 encoded)
+def gen_layout():
+    # create matrix to fit layout
+    placeholder = formatting("~", 1, 0, "center")
+    w, h = len(weekdays)+1, len(hours)+1
+    schedule = [[placeholder.encode("utf8") for x in range(w)] for y in range(h)]
+    # set days & hours in var schedule
+    for i in range(1, len(weekdays)+1):
+        day=formatting(weekdays[i-1], header_color, header_style, header_align)
+        schedule[0][i]=day.encode("utf8")
+    for i in range(len(hours)+1):
+        hour=colors[1]+"|"+formatting(hours[i-1], header_color, header_style, header_align)
+        schedule[i][0]=hour.encode("utf8")
+    index=colors[1]+cell_width*" "+" |"+colors[0]
+    schedule[0][0] = index.encode("utf8")
+    return(schedule)
+
+# formatting
+def formatting(entry, color, style, align):
+    if align == "left":
+        spaces = cell_width-len(entry)-1
+        entry = styles[style]+colors[color]+" "+entry+spaces*" "
+    if align == "center":
+        entry = entry.center(cell_width)
+        entry = styles[style]+colors[color]+entry
+    if align == "right":
+        spaces = cell_width-len(entry)-1
+        entry = styles[style]+colors[color]+spaces*" "+entry+" "
+    return(entry+styles[0]+colors[1]+"|"+colors[0])
+
+# print schedule (decode)
+def print_sched(schedule):
+    os.system('cls' if os.name == 'nt' else 'clear')
+    decoded_sched = []
+    for i in range(len(schedule)):
+        decoded_line = []
+        for j in range(len(schedule[i])):
+             decoded_line.append(schedule[i][j].decode("utf8"))
+        decoded_sched.append(decoded_line)
+    for row in decoded_sched:
+        print(''.join(map(str,row)))
+
+# read saved .csv schedule (utf8 encoded)
+def read_sched(file):
+    file = open(file, "rb")
+    schedule = []
+    for line in file:
+        i = []
+        entries = (line.replace(("\n").encode("utf8"), ("").encode("utf8"))).split((",").encode("utf8"))
+        for j in entries:
+            i.append(bytes(j))
+        schedule.append(i)
+    file.close()
+    return(schedule)
+
+# save function (utf8 encoded)
+def save(schedule, file):
+    with open(file, "wb") as f:
+        for row in schedule:
+            for entry in row:
+                f.write(entry + (",").encode("utf"))
+            f.write(("\n").encode("utf8"))
+
+# add/edit entry in schedule
+def edit(schedule):
+    if len(weekdays)+1 != len(schedule[0]) or len(hours)+1 != len(schedule):
+        print("Imported and configured layouts don't match! Can't edit, only print.")
+        return 0
+    day = menu(weekdays, schedule, 0)
+    hour = menu(hours, schedule, 0)
+    print_sched(schedule)
+    print(styles[1]+"\nType what the activity should be:"+styles[0])
+    entry = input("-> ")
+    if len(entry) > cell_width-2:
+        print("Entry is too long for cell_width!")
+        return 0
+    color = menu(colors, schedule, 1)
+    style = menu(styles, schedule, 1)
+    entry = formatting(entry, color, style, entry_align)
+    schedule[hour+1][day+1] = entry.encode("utf8")
+
+# menu function for opt-lists
+def menu(array, schedule, nopt):
+    while True:
+        print_sched(schedule)
+        print(styles[1]+"\nChoose an option:"+styles[0])
+        if nopt == 1:
+            for i in range(len(array)):
+                print(array[i]+str(i)+array[0])
+
+        elif nopt == 0:
+            for i in range(len(array)):
+                print(styles[1]+str(i)+styles[0]+" - "+array[i])
+        try:
+            opt = int(input("-> "))
+            if 0 <= opt <= len(array):
+                return(opt)
+            else:
+                print("\nOut of range!")
+                input("Try again...")
+        except:
+            print("\nInvalid input! Should be integer.")
+            input("Try again...")
+
+# read arguments
+if (len(sys.argv)) < 2:
+    print(help)
+    sys.exit(0)
+
+# main execution
+mode = sys.argv[1]
+if mode == "new" or mode == "edit":
+    if mode == "new":
+        schedule = gen_layout()
+        print_sched(schedule)
+    elif mode == "edit":
+        try:
+            schedule = read_sched(sys.argv[2])
+        except:
+            print("Invalid input file!")
+    opts = ["print", "edit", "save", "exit"]
+    while True:
+        print_sched(schedule)
+        opt = menu(opts, schedule, 0)
+        if opt == 0:        # print
+            print_sched(schedule)
+        elif opt == 1:      # edit
+            print_sched(schedule)
+            edit(schedule)
+        elif opt == 2:      # save
+            print("Type the name of the schedule file (recommended .txt extension):")
+            file = input("-> ")
+            save(schedule, file)
+        elif opt == 3:      # exit
+            break
+elif mode == "print":
+    try:
+        schedule = read_sched(sys.argv[2])
+    except:
+        print("Invalid input file!")
+    print_sched(schedule)