🚀 go-pugleaf

In the darkest hour on Tue, 15 Nov 2011 03:48:34 +0100,
hubert depesz lubaczewski <depesz@depesz.com> screamed:
> def calculate_hash( filename ):
>     f = open( filename, 'r' )
>     mail = email.message_from_file( f )
>     keep_headers =  ( 'subject', 'from', 'to', 'cc', 'in-reply-to', 'references', 'date', 'message-id', 'content-type', 'mime-version' )
>     for h in mail.keys():
>         if h.lower() not in keep_headers:
>             del mail[ h ]
>     return sha256( mail.as_string() ).hexdigest()
>

W obecnych Pythonach możesz stosować konstrukcję (automagicznie zamknie uchwyt):

def calculate_hash(filename):
    with open(filename) as f:
        mail = .....

f = open(filename, 'r') to to samo co f = open(filename)

> def visit_fn( hash_dict, path, files ):
>     for i in [ 'new', 'tmp' ]:
>         if files.count ( i ) > 0:
>             files.remove( i )
>

Tu lepiej użyć tupli ('new', 'tmp').

>         checksum = calculate_hash( full_path )
>         if not hash_dict.has_key( checksum ):
>             hash_dict[ checksum ] = list()
>         hash_dict[ checksum ].append( full_path )
>
> hash_dict = dict()
>

Alternatywnie (z tym, że trzeba uważać):
hash_dict[checksum] = []
hash_dict = {}

Oprócz konstrukcji:   if not d.has_key(k):
można też używać:     if k not in d:

Dla mnie ta druga jest czytelniejsza.

> walk( sys.argv[1], visit_fn, hash_dict )
>
> for i in hash_dict.keys():
>     if len( hash_dict[i] ) == 1:
>         del hash_dict[i]
>
> for sha in hash_dict.keys():
>     print "# %s" % ( sha )

Alternatywnie, po prostu:   print '#', sha

-- 
[ Artur M. Piwko : Pipen : AMP29-RIPE : RLU:100918 : From == Trap! : SIG:239B ]
[ 09:10:35 user up 12987 days, 21:05,  1 user, load average: 0.09, 0.04, 0.35 ]

 Peace:  In international affairs, a period of cheating between two countries.