From f7af23ec3f6b0c31028327b08240bd5ed95b8a98 Mon Sep 17 00:00:00 2001 From: kaylee Date: Fri, 2 Feb 2024 08:55:12 +0000 Subject: [PATCH] added check function --- uvgotmail.py | 39 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) mode change 100644 => 100755 uvgotmail.py diff --git a/uvgotmail.py b/uvgotmail.py old mode 100644 new mode 100755 index e69de29..3fbbc7e --- a/uvgotmail.py +++ b/uvgotmail.py @@ -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();