server.py

import falcon
import logging
from GhettoYubiHSM import GhettoYubiHSM

FORMAT = '%(asctime)-15s %(levelname)s %(funcName)s(%(lineno)d): %(message)s'
logging.basicConfig(format=FORMAT, level=logging.DEBUG)
LOG = logging.getLogger(__name__)

class EncryptResource:
    """
    Resource to encrypt plaintext and return the ciphertext.
    """

    def on_post(self, req, resp):
        """
        Handle HTTP POST requests.
        Args:
            req: the request object.
            resp: the response object.
        """

        pt = req.stream.read()
        LOG.debug("plaintext: %s" % pt)

        ct = self.encrypt(pt)
        LOG.debug("ciphertext: %s" % ct)

        resp.body = ct

    def encrypt(self, pt):
        """
        This method will "encrypt" the provided plaintext value.
        """
        hsm = GhettoYubiHSM()
        return hsm.encrypt(pt)

class DecryptResource:
    """
    Resource to decrypt ciphertext and return the plaintext.
    """

    def on_post(self, req, resp):
        """
        Handle HTTP POST requests.
        Args:
            req: the request object.
            resp: the response object.
        """

        ct = req.stream.read()
        LOG.debug("ciphertext: %s" % ct)

        pt = self.decrypt(ct)
        LOG.debug("plaintext: %s" % pt)

        resp.body = pt

    def decrypt(self, ct):
        """
        This method will "decrypt" the provided ciphertext value.
        """
        hsm = GhettoYubiHSM()
        return hsm.decrypt(ct)


app = falcon.API()
app.add_route('/decrypt', DecryptResource())
app.add_route('/encrypt', EncryptResource())