1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
# Copyright (C) 2012 by Brian Neal.
# This file is part of Py-Enigma, the Enigma Machine simulation.
# Py-Enigma is released under the MIT License (see License.txt).
"""Contains factory functions for creating rotors and reflectors."""
from . import RotorError
from .rotor import Rotor
from .data import ROTORS, REFLECTORS
def create_rotor(model, ring_setting=0):
"""Factory function to create and return a rotor of the given model name."""
if model in ROTORS:
data = ROTORS[model]
return Rotor(model, data['wiring'], ring_setting, data['stepping'])
raise RotorError("Unknown rotor type: %s" % model)
def create_reflector(model):
"""Factory function to create and return a reflector of the given model
name.
"""
if model in REFLECTORS:
return Rotor(model, wiring=REFLECTORS[model])
raise RotorError("Unknown reflector type: %s" % model)
|
