added check function

This commit is contained in:
kaylee 2024-02-02 08:55:12 +00:00
parent 353eba5605
commit f7af23ec3f
1 changed files with 39 additions and 0 deletions

39
uvgotmail.py Normal file → Executable file
View File

@ -0,0 +1,39 @@
#!/usr/bin/env python3
import argparse
import sys
import os
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--daemon", help="run in daemon mode (put this at end of .bashrc)", action="store_true")
parser.add_argument("--config", help="specify config file in muttrc format")
parser.add_argument("--check", help="run in check mode (put this in PS1)", action="store_true")
parser.add_argument("--no-count", help="don't display how many unread messages there are", action="store_false")
parser.add_argument("--mailbox-char", help="specify a character instead of the mailbox emoji",default="📬")
args = parser.parse_args()
global mailchar, countmail;
mailchar = args.mailbox_char
countmail = args.no_count;
if(args.daemon == args.check):
parser.print_help()
sys.exit(0)
if(args.daemon):
daemon()
else:
check()
def check():
filename = '/tmp/uvgotmail/'+os.getlogin()+'/unread'
if os.path.exists(filename):
with open(filename) as f:
unread = int(f.read());
if unread == 0:
sys.exit(0)
pre = ""
if(unread > 1 and countmail):
pre = "("+str(unread)+") "
print(pre+mailchar,end=' ')
main();