diff --git a/motd.py b/motd.py index b09d4c2..444b195 100755 --- a/motd.py +++ b/motd.py @@ -1,9 +1,12 @@ #!/bin/python +from functools import cache import os -import ueberzug.lib.v0 as ueberzug -import curses +import random +#import ueberzug.lib.v0 as ueberzug +import subprocess import time +import fnmatch from curses.textpad import Textbox, rectangle modpath = os.path.dirname(__file__) @@ -22,11 +25,49 @@ minTermW = 200 DIR=modpath + '/waifus' #print(len([name for name in os.listdir(DIR) if os.path.isfile(os.path.join(DIR, name))])) +#filter waifus +waifulist = os.listdir(DIR) +waifus_raw = fnmatch.filter(waifulist, '*.256.png') +waifus = list(waifus_raw) +waifuindex = random.randint(1, len(waifus)) + + + + + #print(modpath) +####################### + +#If term Kitty +print(len(waifus)) + +if random.randint(0,1) == 0: + if os.path.getsize(cachefile) != 0: + f = open(cachefile, 'r') + waifuindex = f.read() + print(int(waifuindex)) + #print(len(waifus)) + #Else Add File + ############### + if int(waifuindex) < len(waifus): + waifu = waifus[int(waifuindex)] + waifuindex = int(waifuindex) + 1 + f = open(cachefile, 'w') + f.write(str(waifuindex)) + else: + waifuindex = random.randint(1, len(waifus)) + f = open(cachefile, 'w') + f.write(str(waifuindex)) + waifu = waifus[int(waifuindex)] +else: + waifu = waifus[int(waifuindex)] +####################### + +#Declare static methods @staticmethod def showImage(): - print("showImage") + subprocess.run(["/usr/bin/kitty", "icat", DIR+"/"+waifu]) def showUname(unameArg): os.system("/usr/bin/uname " + unameArg) @@ -36,10 +77,9 @@ def showUname(unameArg): def showUptime(uptimeArg): os.system("/usr/bin/uptime " + uptimeArg) -def doLolcat(): +#def doLolcat(): +# if __name__ == '__main__': +# showUname(unameArg) +# showUptime(uptimeArg) - - -if __name__ == '__main__': - showUname(unameArg) - showUptime(uptimeArg) +showImage()