FileManager

Complexity

Total Complexity

6

Size/Duplication

Total Lines	39
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	1

4 Methods

Rating	Name	Duplication	Size	Complexity
A	retrieve_file()	0	11	2
A	store_file()	0	11	2
A	delete_file()	0	7	1
A	__init__()	0	3	1

"""Package defining Redis file manager

.. Authors:
    Philippe Dessauw
    philippe.dessauw@nist.gov

.. Sponsor:
    Alden Dima
    alden.dima@nist.gov
    Information Systems Group
    Software and Systems Division
    Information Technology Laboratory
    National Institute of Standards and Technology
    http://www.nist.gov/itl/ssd/is
"""
import base64
from os import remove
import redis


class FileManager(object):
    """Redis file manager
    """

    def __init__(self, host="127.0.0.1", port=6379, db=0):
        self.server = redis.StrictRedis(host, port, db)
        self.hashmap_name = "fman"

    def retrieve_file(self, filename):
        """Retrieve from redis hashmap

        Args
            filename (str): Filename to retrieve
        """
        b64_hash = self.server.hget(self.hashmap_name, filename)
        data = base64.b64decode(b64_hash)

        with open(filename, 'wb') as zip_file:
            zip_file.write(data)

    def store_file(self, filename):
        """Store file to redis hashmap

        Args
            filename (str): Filename to store
        """
        with open(filename, 'rb') as zip_file:
            b64_hash = base64.b64encode(zip_file.read())

        self.server.hset(self.hashmap_name, filename, b64_hash)
        remove(filename)

    def delete_file(self, filename):
        """Delete file from redis hashmap

        Args
            filename (str): Filename to delete
        """
        self.server.hdel(self.hashmap_name, filename)