Das deutsche Python-Forum

Code: Alles auswählen

("F:", "579EF173")
("G:", "878FH189")
("H:", "8G6T9AF8")

Code: Alles auswählen

from subprocess import Popen, PIPE
import re 

VOLUME_LETTERS = ["A:","B:","C:","D:","E:","F:","G:","H:","I:","J:","K:","L:","M:","N:","O:","P:","Q:","R:","S:","T:","U:","V:","W:","X:","Y:","Z:"]

def get_volume_serial_number_and_letter():
    volume_letter_to_serial_number = {}
    for volume_letter in VOLUME_LETTERS:
        process = Popen(['vol', volume_letter], stdout=PIPE, stderr=PIPE, shell=True)
        stdout, stderr = process.communicate()
        decoded_stdout = stdout.decode('windows-1252')
        search_index_start = re.search("Volumeseriennummer: ", decoded_stdout)
        try:
            volume_letter_to_serial_number[volume_letter] = decoded_stdout[search_index_start.span()[1]:search_index_start.span()[1] + 9] #serial number is 9 chars long
        except AttributeError:
            volume_letter_to_serial_number[volume_letter] = None
    return volume_letter_to_serial_number


print(get_volume_serial_number_and_letter())

Code: Alles auswählen

for i in range(26):
    volume = f"{chr(ord('A') + i)}:"
    print(volume)

Code: Alles auswählen

import string
print([f"{char}:" for char in string.ascii_uppercase])

Code: Alles auswählen

import string

def volumeletters():
    for item in string.ascii_uppercase:
        yield f"{item}:"

for volumeletter in volumeletters():
    print(volumeletter)

Code: Alles auswählen

import wmi

for device in wmi.WMI().Win32_LogicalDisk():
    name = device.wmi_property('Name').value
    serial_number = device.wmi_property('VolumeSerialNumber').value
    print(name, serial_number)

Code: Alles auswählen

import wmi
import argon2


def get_volume_serial_number_and_letter_wmi():
        volume_letter_to_serial_number = dict()
        for volume in wmi.WMI().Win32_LogicalDisk():
            name = volume.wmi_property('Name').value
            serial_number = volume.wmi_property('VolumeSerialNumber').value
            volume_letter_to_serial_number[name] = serial_number
        return volume_letter_to_serial_number

def get_difference_of_dicts(dict1, dict2):
    difference = {}
    for key in dict1.keys():
        if key not in dict2.keys():
            difference[key] = dict1[key]
    return difference  
     


def main():
      while True:
        print("Was möchtest du tun?: ")
        choice = int(input("1 - Einstellungen setzen \n2 - Einstellungen prüfen\n\n> "))
        if choice == 1: 
            input("Entferne alle USB-Sticks aus dem USB-Hub. Drücke [Enter], sobald alle USB-Sticks entfernt sind\n")
            volume_letters_and_serial_without_usb_sticks = get_volume_serial_number_and_letter_wmi()
            input("Stecke nun die USB-Sticks in der gewünschten Reihenfolge in den USB-Hub und drücke anschließend [Enter]\n")
            volume_letters_and_serial_with_usb_sticks = get_volume_serial_number_and_letter_wmi()
            difference = get_difference_of_dicts(volume_letters_and_serial_with_usb_sticks, volume_letters_and_serial_without_usb_sticks)
            for key, value in difference.items():
                difference[key] = argon2.PasswordHasher().hash(value)
            print(difference)
        if choice == 2:
            volume_letters_and_serial_with_usb_sticks = get_volume_serial_number_and_letter_wmi()
            difference2 = get_difference_of_dicts(volume_letters_and_serial_with_usb_sticks, volume_letters_and_serial_without_usb_sticks)
            if difference2:
                for key, value in difference2.items():
                    if key in difference.keys():
                        if argon2.PasswordHasher().verify(difference[key],difference2[key]):
                            print("YES")
                        else:
                            print("NO")
                    else:
                        print("NO")
            else:
                print("NO")

main()

Code: Alles auswählen

def get_difference_of_dicts(input, keys_to_be_removed):
    return {key: input[key] for key in set(input) - set(keys_to_be_removed)}

Code: Alles auswählen

#!/usr/bin/env python3
import argon2
import wmi


def get_property(volume, name):
    return volume.wmi_property(name).value


def get_volume_serial_numbers():
    return dict(
        (
            get_property(volume, "Name"),
            get_property(volume, "VolumeSerialNumber"),
        )
        for volume in wmi.WMI().Win32_LogicalDisk()
    )


def subtract_mappings(mapping_a, mapping_b):
    return {key: mapping_a[key] for key in mapping_a.keys() - mapping_b.keys()}


def configure():
    input(
        "Entferne alle USB-Sticks aus dem USB-Hub. Drücke [Enter], sobald"
        " alle USB-Sticks entfernt sind\n"
    )
    volume_letters_and_serial_without_usb_sticks = get_volume_serial_numbers()
    input(
        "Stecke nun die USB-Sticks in der gewünschten Reihenfolge in den"
        " USB-Hub und drücke anschließend [Enter]\n"
    )
    volume_letters_and_serial_with_usb_sticks = get_volume_serial_numbers()
    difference = {
        letter: argon2.PasswordHasher().hash(serial)
        for letter, serial in subtract_mappings(
            volume_letters_and_serial_with_usb_sticks,
            volume_letters_and_serial_without_usb_sticks,
        ).items()
    }
    print(difference)
    return volume_letters_and_serial_without_usb_sticks, difference


def verify(volume_letters_and_serial_without_usb_sticks, difference):
    volume_letters_and_serial_with_usb_sticks = get_volume_serial_numbers()
    difference2 = subtract_mappings(
        volume_letters_and_serial_with_usb_sticks,
        volume_letters_and_serial_without_usb_sticks,
    )
    if difference2:
        for letter, serial in difference2.items():
            if letter in difference:
                if argon2.PasswordHasher().verify(difference[letter], serial):
                    print("YES")
                else:
                    print("NO")
            else:
                print("NO")
    else:
        print("NO")


def main():
    volume_letters_and_serial_without_usb_sticks, difference = configure()

    while True:
        print("Was möchtest du tun?")
        while True:
            try:
                choice = int(
                    input(
                        "1 - Einstellungen setzen\n"
                        "2 - Einstellungen prüfen\n\n> "
                    )
                )
                break
            except ValueError:
                print("Fehler: Bitte eine ganze Zahl eingeben!")

        if choice == 1:
            (
                volume_letters_and_serial_without_usb_sticks,
                difference,
            ) = configure()

        elif choice == 2:
            verify(volume_letters_and_serial_without_usb_sticks, difference)

        else:
            print("Fehler: Ungültige Auswahl!")


if __name__ == "__main__":
    main()

Code: Alles auswählen

import wmi
import argon2
import json
import pathlib

CONFIGURATION_FILE = pathlib.Path("TEST.txt")

def get_property(volume, name):
    return volume.wmi_property(name).value

def get_usb_serial_and_drive_letter():
    return dict((get_property(volume, "Name"),get_property(volume, "VolumeSerialNumber")) for volume in wmi.WMI().Win32_LogicalDisk() if volume.DriveType == 2)

def write_to_config_file(configuration):
    with open (CONFIGURATION_FILE, 'w', encoding='UTF-8') as output_file:
        output_file.write(json.dumps(configuration))

def read_config_file():
    if CONFIGURATION_FILE.exists():
        with open (CONFIGURATION_FILE, 'r', encoding='UTF-8') as input_file:
            return json.loads(input_file.read())
    else:
        return 

    
def verify_config(saved_config, current_connected_usb_volumes):
    try:
        results = [argon2.PasswordHasher().verify(saved_config[letter], current_connected_usb_volumes[letter]) for letter in saved_config.keys()]
    except:
        return False
    if results:
        return all(results)
    else:
        return False

def main():
    while True:
        print("Was möchtest du tun?")
        while True:
            try:
                choice = int(input("1 - Einstellungen setzen\n2 - Einstellungen prüfen\n\n> "))
                break
            except ValueError:
                print("Fehler: Bitte eine ganze Zahl eingeben!")
        if choice == 1:
            hashed_current_connected_usb_volumes = {letter: argon2.PasswordHasher().hash(serial) for letter, serial in get_usb_serial_and_drive_letter().items()}
            if hashed_current_connected_usb_volumes:
                write_to_config_file(hashed_current_connected_usb_volumes)
            else:
                print("Keine USB-Stick gefunden. Vergewissere dich, dass USB-Sticks eingesteckt sind und versuche es erneut.")
        if choice == 2:
            saved_config = read_config_file()
            if saved_config:
                current_connected_usb_volumes = get_usb_serial_and_drive_letter()
                is_verified = verify_config(saved_config, current_connected_usb_volumes)
                if is_verified:
                    print("Die USB-Sticks sind in der richtigen Reihenfolge eingesteckt")
                else:
                    print("Die USB-Sticks sind nicht in der richtigen Reihenfolge eingesteckt")
            else:
                print("Es wurde keine Konfigurationsdatei gefunden.")
                
if __name__ == "__main__":
    main()

Code: Alles auswählen

results = [argon2.PasswordHasher().verify(saved_config[letter], current_connected_usb_volumes[letter]) for letter in current_connected_usb_volumes.keys()]

Code: Alles auswählen

results = [argon2.PasswordHasher().verify(saved_config[letter], current_connected_usb_volumes[letter]) for letter in saved_config.keys()]

Code: Alles auswählen

def get_usb_serial_and_drive_letter():
    return {
        get_property(volume, "Name"): get_property(volume, "VolumeSerialNumber")
        for volume in wmi.WMI().Win32_LogicalDisk()
        if volume.DriveType == 2
    }

Code: Alles auswählen

def get_usb_serial_and_drive_letter():
    return {
        get_property(volume, "Name"): get_property(
            volume, "VolumeSerialNumber"
        )
        for volume in wmi.WMI().Win32_LogicalDisk()
        if volume.DriveType == 2
    }

Code: Alles auswählen

import wmi
import argon2
import json


CONFIGURATION_FILE = "TEST.txt"

def get_property(volume, name):
    return volume.wmi_property(name).value

def get_usb_serial_and_drive_letter():
    return {get_property(volume, "Name") : get_property(volume, "VolumeSerialNumber") for volume in wmi.WMI().Win32_LogicalDisk() if volume.DriveType == 2}

def write_to_config_file(configuration):
    with open (CONFIGURATION_FILE, 'w', encoding='UTF-8') as output_file:
        output_file.write(json.dumps(configuration))

def read_config_file():
    try:
        with open (CONFIGURATION_FILE, 'r', encoding='UTF-8') as input_file:
            return json.loads(input_file.read())
    except FileNotFoundError:
        return {}
    except json.JSONDecodeError:
        return {}

    
def verify_config(saved_config, current_connected_usb_volumes):
    try:
        results = [argon2.PasswordHasher().verify(saved_config[letter], current_connected_usb_volumes[letter]) for letter in saved_config.keys()]
        if results:
            return all(results)
        else:
            return False
    except argon2.exceptions.VerifyMismatchError:
        return False
    except argon2.exceptions.VerificationError:
        return False
    
def main():
    while True:
        while True:
            print("Was möchtest du tun?")
            try:
                choice = int(input("1 - Einstellungen setzen\n2 - Einstellungen prüfen\n\n> "))
                break
            except ValueError:
                print("Fehler: Bitte eine ganze Zahl eingeben!\n")
        if choice == 1:
            hashed_current_connected_usb_volumes = {letter : argon2.PasswordHasher().hash(serial) for letter, serial in get_usb_serial_and_drive_letter().items()}
            if hashed_current_connected_usb_volumes:
                write_to_config_file(hashed_current_connected_usb_volumes)
            else:
                print("Keine USB-Stick gefunden. Vergewissere dich, dass USB-Sticks eingesteckt sind und versuche es erneut.\n")
        if choice == 2:
            saved_config = read_config_file()
            if saved_config:
                current_connected_usb_volumes = get_usb_serial_and_drive_letter()
                is_verified = verify_config(saved_config, current_connected_usb_volumes)
                if is_verified:
                    print("Verifikation erfolgreich.\n")
                else:
                    print("Verifikation fehlgeschlagen\n")
            else:
                print("Es wurde keine Konfigurationsdatei gefunden, oder die Konfigurationsdatei ist fehlerhaft.\n")
                
if __name__ == "__main__":
    main()