X-Git-Url: https://git.deb.at/w?a=blobdiff_plain;f=contrib%2Fmail2abook.py;fp=contrib%2Fmail2abook.py;h=8efcedac97d7ae490c6ac67af06f4c431e06644d;hb=9d28f71348269e7075d0ff5c0b7ae4e05f358304;hp=0000000000000000000000000000000000000000;hpb=c0b0c6c7a0971525558239e544212589fbbac58b;p=pkg%2Fabook.git diff --git a/contrib/mail2abook.py b/contrib/mail2abook.py new file mode 100755 index 0000000..8efceda --- /dev/null +++ b/contrib/mail2abook.py @@ -0,0 +1,438 @@ +#!/usr/bin/env python + +# NOTE: starting from abook version 0.4.15 it is often more convenient +# to use command line option "--add-email" instead of this script + +# mail2abook.py version 0.1 +# by Jaakko Heinonen +# based on Moritz Moeller-Herrmann's mail2muttalias.py + +# Description: +# this script reads all mail adresses from a piped mailfile and +# offers to write them to the ABOOKFILE specified in the script +# Append the following lines or similiar to your muttrc: +# macro pager A |'path to script'/mail2abook.py\n + +# Then press A in pager mode to add an address to your abook addressbook + +# Here's the original copyright information: + +# RELEASE 0.5 +## +# Needs python 1.5 and cursesmodule 1.4 by andrich@fga.de (Oliver Andrich) +# available somewhere on http://www.python.org or directly from +# http://andrich.net/python/selfmade.html +# +# Copyright by Moritz Moeller-Herrmann +# +# Homepage (check for changes before reporting a bug) +# http://webrum.uni-mannheim.de/jura/moritz/mail2muttalias.html +# +# Use this any way you want. Please inform me of any error/improvement +# and leave the copyright notice untouched. +# No warranties, as this is my first program :-) +# Works for me on Linux with Python 1.5+ +# +# Thanks to Josh Hildebrand for some improvements +## + +# Known bugs: +# If name containing @ is quoted("") before a mail adress, +# the program will not catch the whole name, +# but who would alias such an idiot? +# If you get a problem with "unknown variable / keyword error" +# with KEYUP and KEYDOWN, either get a newer pythoncurses module +# or change them to lowercase. +# +# Probably some more, mail me if you find one! + +import re +import string, sys, os +import curses, traceback +import locale + +locale.setlocale(locale.LC_ALL,os.environ["LANG"]) + +ABOOKFILE=os.environ["HOME"] + "/.abook/addressbook" + +try : + testcurses = curses.KEY_UP +except NameError: + print "Your pythoncurses module is old. Please upgrade to 1.4 or higher." + print "Alternative: Modify the script. Change all 6 occurences of curses.KEY_UP" + print "and similiar to lowercase." + +try : testAF = ABOOKFILE #test if ALIASFILE was configured in script +except NameError: + try: ALIASFILE=os.environ["ABOOKFILE"] #test is environment MUTTALIASFILE was set + except KeyError: + print "Please specify ABOOKFILE at beginning of script \nor set environment variable ABOOKFILE" + print "Aborting ..." + sys.exit() + + + +###Thanks for the following go to Michael P. Reilly +if not sys.stdin.isatty(): # is it not attached to a terminal? + file = sys.stdin.read() + sys.stdin = _dev_tty = open('/dev/tty', 'w+') + # close the file descriptor for stdin thru the posix module + os.close(0) + # now we duplicate the opened _dev_tty file at file descriptor 0 (fd0) + # really, dup creates the duplicate at the lowest numbered, closed file + # descriptor, but since we just closed fd0, we know we will dup to fd0 + os.dup(_dev_tty.fileno()) # on UNIX, the fileno() method returns the + # file object's file descriptor +else: # is a terminal + print "Please use as a pipe!" + print "Aborting..." + sys.exit() +# now standard input points to the terminal again, at the C level, not just +# at the Python level. + + +print "Looking for mail adresses, this may take a while..." + + +#### define functions + + +class screenC: + "Class to create a simple to use menu using curses" + def __init__(self): + import curses, traceback + self.MAXSECT = 0 + self.LSECT = 1 + # Initialize curses + self.stdscr=curses.initscr() + # Turn off echoing of keys, and enter cbreak mode, + # where no buffering is performed on keyboard input + curses.noecho() + curses.cbreak() + + # In keypad mode, escape sequences for special keys + # (like the cursor keys) will be interpreted and + # a special value like curses.KEY_LEFT will be returned + self.stdscr.keypad(1) + + def titel(self,TITEL="Title - test",HELP="Use up and down arrows + Return"): + "Draws Title and Instructions" + self.stdscr.addstr(0,0,TITEL,curses.A_UNDERLINE) # Title + Help + self.stdscr.addstr(self.Y -2 ,0,HELP,curses.A_REVERSE) + + def refresh(self): + self.stdscr.refresh() + + def size(self): + "Returns screen size and cursor position" + #Y, X = 0, 0 + self.Y, self.X = self.stdscr.getmaxyx() + #self.y, self.x = 0, 0 + self.y, self.x = self.stdscr.getyx() + #print Y, X + return self.Y, self.X, self.y, self.x + + + def showlist(self,LISTE,LSECT=1): + "Analyzes list, calculates screen, draws current part of list on screen " + s = self.Y -3 #space on screen + self.MAXSECT=1 + while len(LISTE) > self.MAXSECT * s : # how many times do we need the screen? + self.MAXSECT = self.MAXSECT +1 + + if self.LSECT > self.MAXSECT: #check for end of list + self.LSECT = LSECT -1 + + if self.LSECT <= 0: # + self.LSECT =1 + + if len(LISTE) <= s: + self.LISTPART=LISTE + + else : + self.LISTPART=LISTE[s * ( self.LSECT -1 ) : s * self.LSECT ] # part of the List is shown + + self.stdscr.addstr(self.Y -2, self.X -len(`self.LSECT`+`self.MAXSECT`) -5, "(" + `self.LSECT` + "/" + `self.MAXSECT` + ")") + #if len(LISTE) > self.Y - 3: + # return 1 + for i in range (1, self.Y -2): # clear screen between title and help text + self.stdscr.move(i , 0) + self.stdscr.clrtoeol() + for i in range (0,len(self.LISTPART)): # print out current part of list + Item = self.LISTPART[i] + self.stdscr.addstr(i +1, 0, Item[:self.X]) + + def getresult(self,HOEHE): + "Get Result from cursor position" + RESULT= self.LISTPART[(HOEHE -1)] + return RESULT + + def showresult(self, HOEHE, DICT={}): + "Look up result to show in dictionary if provided, return list member otherwise" + if DICT=={}: + return self.getresult(HOEHE) + else : + return string.join(DICT[self.getresult(HOEHE)], " || ") + + + + def menucall(self, LISTE, DICT={}, TITEL="",HELP="Use up and down arrows, Return to select"): + "Takes a list and offers a menu where you can choose one item, optionally, look up result in dictionary if provided" + REFY=1 + self.__init__() + self.size() + self.titel(TITEL,HELP) + self.showlist(LISTE) + self.refresh() + self.stdscr.move(1,0) + while 1: # read Key events + c = self.stdscr.getch() + self.size() + + #if c == curses.KEY_LEFT and self.x > 0: + # self.stdscr.move(self.y, self.x -1); REFY = 1 # REFY == refresh: yes + + #if c == curses.KEY_RIGHT and self.x < self.X -1: + # #if x < 4 and LENGTH - ZAHLY > y - 1: + # self.stdscr.move(self.y, self.x + 1); REFY = 1 + + if c == curses.KEY_UP or c == 107: #up arrow or k + if self.y > 1: + self.stdscr.move(self.y -1, self.x); REFY = 1 + else : + self.LSECT=self.LSECT-1 + self.showlist(LISTE,self.LSECT) + self.stdscr.move(len(self.LISTPART), 0) + REFY = 1 + + if c == curses.KEY_DOWN or c == 106: #down arrow or j + + if self.y < len(self.LISTPART) : + self.stdscr.move(self.y +1, self.x); REFY = 1 + + else : + self.LSECT=self.LSECT+1 + self.showlist(LISTE,self.LSECT) + self.stdscr.move(1,0) + REFY = 1 + + if c == curses.KEY_PPAGE: + self.LSECT=self.LSECT-1 + self.showlist(LISTE,self.LSECT) + self.stdscr.move(1, 0) + REFY = 1 + + if c == curses.KEY_NPAGE: + self.LSECT=self.LSECT+1 + self.showlist(LISTE,self.LSECT) + self.stdscr.move(1,0) + REFY = 1 + + if c == curses.KEY_END: + self.LSECT=self.MAXSECT + self.showlist(LISTE,self.LSECT) + self.stdscr.move(1,0) + REFY = 1 + + if c == curses.KEY_HOME: + self.LSECT=1 + self.showlist(LISTE,self.LSECT) + self.stdscr.move(1,0) + REFY = 1 + + + if c == 10 : # \n (new line) + ERG = self.getresult(self.y ) + self.end() + return ERG + + if c == 113 or c == 81: # "q or Q" + self.printoutnwait("Aborted by user!") + self.end() + sys.exit() + return 0 + + if REFY == 1: + REFY = 0 + self.size() + self.stdscr.move(self.Y -1, 0) + self.stdscr.clrtoeol() + self.stdscr.addstr(self.Y -1, 0, self.showresult(self.y,DICT)[:self.X -1 ], curses.A_BOLD) + self.stdscr.move(self.y, self.x) + self.refresh() + + def end(self): + "Return terminal" + # In the event of an error, restore the terminal + # to a sane state. + self.Y, self.X, self.y, self.x = 0, 0, 0, 0 + self.LISTPART=[] + self.stdscr.keypad(0) + curses.echo() + curses.nocbreak() + curses.endwin() + #traceback.print_exc() + + def input(self, promptstr): + "raw_input equivalent in curses, asks for Input and returns it" + self.size() + curses.echo() + self.stdscr.move(0,0) + self.stdscr.clear() + self.stdscr.addstr(promptstr) + self.refresh() + INPUT=self.stdscr.getstr() + curses.noecho() + return INPUT + + + def printoutnwait(self, promptstr): + "Print out Text, wait for key" + curses.noecho() + self.stdscr.move(0,0) + self.stdscr.clear() +# The new Mutt client pauses after running the script already. No reason +# to pause twice. -Josh +# self.stdscr.addstr(promptstr+"\n\n(press key)") +# self.refresh() +# c = self.stdscr.getch()# read Key events + + + +def listrex (str, rgx): # Return a list of all regexes matched in string + "Search string for all occurences of regex and return a list of them." + result = [] + start = 0 # set counter to zero + ende =len (str) #set end position + suchadress = re.compile(rgx,re.LOCALE)#compile regular expression, with LOCALE + while 1: + einzelerg = suchadress.search(str, start,ende) #create Match Object einzelerg + if einzelerg == None:#until none is found + break + result.append(einzelerg.group()) #add to result + start = einzelerg.end() + return result + +def strrex (str): # Return first occurence of regular exp as string + "Search string for first occurence of regular expression and return it" + muster = re.compile(r"?", re.LOCALE) #compile re + matobj = muster.search(str) #get Match Objekt from string + if muster.search(str) == None: #if none found + return "" + return matobj.group() #return string + +def stripempty (str):#Looks for all empty charcters and replaces them with a space + "Looks for all empty charcters and replaces them with a space,kills trailing" + p = re.compile( "\s+") #shorten + shrt = p.sub(" ", str) + q = re.compile("^\s+|\s+$") #strip + strp = q.sub("", shrt) + return strp + +def getmailadressfromstring(str): + "Takes str and gets the first word containing @ == mail adress" + StringSplit=string.split(str) + for i in range(len(StringSplit)): + if "@" in StringSplit[i]: + return StringSplit[i] + return None + +### main program + +OCCLIST = listrex(file, '"?[\s\w\ö\ä\ü\-\ß\_.]*"?\s*?')#get list, RE gets all Email adresses + prepending words + +if OCCLIST: + print len(OCCLIST),"possible adresses found!." +else: + print"ERROR, no mails found" + sys.exit() + + +for i in range(len(OCCLIST)): #strip all whitespaces + trailing from each list member + OCCLIST[i] = string.strip(OCCLIST [i]) + + +OCCDIC={} # Dictionary created to avoid duplicates +for i in range(len(OCCLIST)): # iterate over + d = OCCLIST[i] + Mail = getmailadressfromstring(OCCLIST[i]) + #strrex(OCCLIST[i]) #Mailadresse + Schnitt = - len(Mail) #cut off mail adress + Mail = string.replace(Mail, "<", "")#remove <> + Mail = string.replace(Mail, ">", "") + Name = string.replace (stripempty (d[:Schnitt]), '"', '') #leaves name + if not OCCDIC.get(Mail): # if new Emailadress + Liste = [] # create list for names + Liste.append(Name) # append name + OCCDIC[Mail] = Liste # assign list to adress + else : + Liste = OCCDIC[Mail] # otherwise get list + Liste.append(Name) # append name to list of names + OCCDIC[Mail] = Liste # and assign + + +KEYS = OCCDIC.keys() #iterate over dictionary, sort names + #KEYS are all the adresses + +for z in range( len(KEYS) ): + NAMLIST = OCCDIC[KEYS[z]] # Get list of possible names + d = {} # sort out duplicates and + # remove bad names + empty strings from adresses + for x in NAMLIST: + if x in ["", "<"]: continue + d[x]=x + NAMLIST = d.values() + NAMLIST.sort() # sort namelist alphabetically + print z, KEYS[z], "had possible names:", NAMLIST # Debugging output + OCCDIC[KEYS[z]] = NAMLIST # + +print "\n" + +###sorting + +def Comparelength(x, y): + "Compare number of names in OCCDIC, if equal sort alphabetically." + if len(OCCDIC[y]) == len(OCCDIC[x]): + return cmp(x, y) + if len(OCCDIC[y]) < len(OCCDIC[x]): + return -1 + else: + return 1 + +KEYS.sort(Comparelength) # Keys sort + +###menu + +ScreenObject=screenC() # initialize curses menu +try: + ZIELADRESS = ScreenObject.menucall(KEYS, OCCDIC, "Choose adress to alias") + if OCCDIC[ZIELADRESS]: + LISTNAM=["***ENTER own NAME"] #add option to edit name + LISTNAM= LISTNAM + OCCDIC[ZIELADRESS] + ZIELNAME = ScreenObject.menucall(LISTNAM, {}, ZIELADRESS + " has which of the possible names?") + # empty Dictionary {} means show list member itself, not looked up result + else : ZIELNAME="" +except: + T=ScreenObject.size() + ScreenObject.end() +# traceback.print_exc() # Uncomment for curses debugging info +# print T + sys.exit() + +### enter new names/aliases + +if ZIELNAME == "***ENTER own NAME" or ZIELNAME == "": + ZIELNAME = ScreenObject.input(`ZIELADRESS` + " = " + `OCCDIC[ZIELADRESS]` + "\n" + `ZIELADRESS` + " gets which name? ") + +if ZIELNAME == "": + ZIELNAME = "No name" + +WRITEALIAS = "\n[]\nname=" + ZIELNAME + "\nemail=" + ZIELADRESS + "\n\n" + +f = open(ABOOKFILE, "a") +f.write(WRITEALIAS) +f.close() + +ScreenObject.printoutnwait("Item was added to "+ ABOOKFILE + "\nProgam terminated") +ScreenObject.end() +