added check function
This commit is contained in:
parent
353eba5605
commit
f7af23ec3f
|
@ -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();
|
Loading…
Reference in New Issue