memegen.services.link

Complexity

Total Complexity

6

Size/Duplication

Total Lines	28
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	LinkService.encode()	0	7	2
A	LinkService.__init__()	0	3	1
A	LinkService.decode()	0	8	3

import base64

from ._base import Service


class LinkService(Service):

    def __init__(self, template_store, **kwargs):
        super().__init__(**kwargs)
        self.template_store = template_store

    @staticmethod
    def encode(key, path):
        slug = '\t'.join((key, path))
        while len(slug) % 3:
            slug += '\t'
        code = base64.urlsafe_b64encode(slug.encode('utf-8'))
        return code.decode('utf-8')

    def decode(self, code):
        try:
            slug = base64.urlsafe_b64decode(code).decode('utf-8')
            key, path = slug.strip('\t').split('\t')
        except ValueError:
            raise self.exceptions.InvalidMaskedCode
        else:
            return key, path