Das deutsche Python-Forum

Code: Alles auswählen

import hashlib
import sys
import os

import MySQLdb

def main():
    walker = os.walk(os.environ["SYSTEMDRIVE"] + os.sep)
    while True:
        try:
            curdir = walker.next()
        except StopIteration:
            break
        finally:
            for f in curdir[2]:
                insert(curdir[0], f)

def insert(path, fname):
    absolute = path + os.sep + fname
    
    try:
        content = open(absolute, "r").read()
        md5 = hashlib.md5(content).hexdigest()
        sha256 = hashlib.sha256(content).hexdigest()
        sha1 = hashlib.sha1(content).hexdigest()
    except IOError:
        md5 = sha256 = sha1 = ""
        
    cursor.execute("INSERT INTO files VALUES (%s, %s, %s, %s, %s)",
                   (fname, absolute, md5, sha256, sha1))
    connection.commit()    

if __name__ == "__main__":
    try:
        connection = MySQLdb.connect(
            "localhost",
            "root",
            "",
            "system")
    except:
        sys.stdout.write("Can not connect to MySQL database (adress: localhost, user: root, password: '', table: system)")
        sys.exit(0)
    cursor = connection.cursor()
    sys.stdout.write("Running... Pleas be patient")
    main()

Code: Alles auswählen

def main():
    walker = os.walk(os.environ["SYSTEMDRIVE"] + os.sep)
    while True:
        try:
            curdir = walker.next()
        except StopIteration:
            break
        finally:
            for f in curdir[2]:
                insert(curdir[0], f)

Code: Alles auswählen

def main():
    walker = os.walk(os.environ["SYSTEMDRIVE"])
    for dir, _, files in walker:
        for f in files:
            insert(dir, f)

Code: Alles auswählen

  
[...]
try:
    with open(absolute, "r") as f:           
        md5 = hashlib.md5()
        sha256 = hashlib.sha256()    
        sha1 = hashlib.sha1() 
        content = 1           
        while content != "":
            content = f.read(1048576 * 500) # 500 MB
            md5.update(content)
            sha256.update(content)
            sha1.update(content)                
            
    md5 = md5.hexdigest()
    sha256 = sha256.hexdigest()
    sha1 = sha1.hexdigest()
except IOError:
    md5 = sha256 = sha1 = ""
[...]