quests.py

import crypto
import random
from constants import *


messages = [
    "ceci est un message",
    "voici un code",
    "bien le boujour",
    "vive le cryptage",
    "j ai une voiture"
]

words = [
    "vigenere",
    "clef",
    "code",
    "niveau",
    "cryptcrush",
    "atom",
    "python",
    "programmation"
]


def pgcd(a, b):
    if a % b == 0:
        return b
    else:
        return pgcd(b, a % b)


def ri(a):
    c = list(range(1, 27))
    random.shuffle(c)
    b = c.pop(0)
    while pgcd(a, b) != 1:
        b = c.pop(0)
    return b


crypting_method = {
    "affine": [[1, ri(1)],   [3, ri(3)],   [5, ri(5)],   [7, ri(7)],   [9, ri(9)], [11, ri(11)],
               [15, ri(15)], [17, ri(17)], [19, ri(19)], [21, ri(21)], [23, ri(23)]],
    "caesar": [chr(i) for i in range(98, 123)],
    "vigenere": words,
    "polybe": [i for i in range(6, 14)]
}


def choose_crypting_and_message():
    return random.choice(messages), random.choice(list(crypting_method.keys()))


class Message:
    def __init__(self):
        self.__msg, self.crypting_meth = choose_crypting_and_message()
        self.crypted = ""
        self.crypting_keys = random.choice(crypting_method[self.crypting_meth])
        if not isinstance(self.crypting_keys, list):
            self.crypting_keys = [self.crypting_keys]
        self.indices_sent = 0
        print(self.__msg)

    def start(self):
        self.crypted = crypto.crypt(self.crypting_meth, self.__msg, self.crypting_keys)

    def get_clear(self):
        return self.__msg

    def get_indice(self):
        if self.indices_sent == 2:
            self.indices_sent = 3
            if self.do_we_need_two_indices():
                return {'end': "Tu as les indices, maintenant trouve le point final pour décrypter"}
            else:
                return {'key2': self.crypting_keys[1]}
        elif self.indices_sent == 1:
            self.indices_sent = 2
            return {'crypting': self.crypting_meth}
        elif self.indices_sent == 0:
            self.indices_sent = 1
            return {'key': self.crypting_keys[0]}
        else:
            return {}

    def do_we_need_two_indices(self):
        if self.crypting_meth == "affine":
            return False
        else:
            return True

    def decrypt(self, indices, msg):
        if "crypting" in indices.keys() and "key" in indices.keys() and ("key2" in indices.keys() or "end" in indices.keys()):
            keys = [indices["key"]]
            if "key2" in indices.keys():
                keys.append(indices["key2"])
            d = crypto.decrypt(indices["crypting"], self.crypted, keys)
            if d == self.__msg and d == msg:
                return DECRYPT_OK
            return DECRYPT_FAILED
        return DECRYPT_NOT_ENOUGH_INDICES


class Indice:
    def __init__(self, content):
        self.content = content