added check function
This commit is contained in:
		
							parent
							
								
									353eba5605
								
							
						
					
					
						commit
						f7af23ec3f
					
				
							
								
								
									
										39
									
								
								uvgotmail.py
									
									
									
									
									
										
										
										Normal file → Executable file
									
								
							
							
						
						
									
										39
									
								
								uvgotmail.py
									
									
									
									
									
										
										
										Normal file → Executable 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();
 | 
				
			||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user