summaryrefslogtreecommitdiff
path: root/ricky/pbpattern/params.py
blob: 8c3a8855d01e629b706d9ab4f6f2d5e127a5f16b (plain)
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
from ricky.params import Params as _Params
from ricky.param.username import Username
from ricky.param.imageurl import PbageUrl
from ricky.param.enum import Enum
from ricky.config import PATTERN_URL_BASE, PBPATTERN_URL


class Params(_Params):
    def __init__(self):
        super(Params, self).__init__(
           Username(name="username", required=False),
           PbageUrl(name="image_url", required=True),
           Enum(
                name="pattern_url",
                required=True,
                options=self._get_pattern_urls()
           )
        )
        self._url = PBPATTERN_URL

    def _get_pattern_urls(self):
        return set(
            ["%s/img/%s.png" % (PATTERN_URL_BASE, i) for i in xrange(0, 97)] +
            ["%s/img/a%s.png" % (PATTERN_URL_BASE, i) for i in xrange(1, 42)]
        )