File: standard.py

package info (click to toggle)
python-redmine 2.5.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 520 kB
  • sloc: python: 3,785; makefile: 5
file content (98 lines) | stat: -rw-r--r-- 3,617 bytes parent folder | download
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
"""
Defines standard Redmine resources managers.
"""

from . import ResourceManager
from .. import exceptions


class ProjectManager(ResourceManager):
    def __getattr__(self, attr):
        if attr in ('close', 'reopen', 'archive', 'unarchive'):
            if self.redmine.ver is not None and self.redmine.ver < (5, 0, 0):
                raise exceptions.VersionMismatchError(f'Project {attr}')

            return lambda resource_id: self.redmine.engine.request(
                'put', f'{self.redmine.url}{self.resource_class.query_one.format(resource_id)[:-5]}/{attr}.json')

        raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{attr}'")


class IssueManager(ResourceManager):
    def copy(self, issue_id, link_original=True, include=(), **fields):
        fields['_copy'] = {'copy_from': issue_id}

        if link_original:
            fields['_copy']['link_copy'] = '1'

        if include is not None:
            for i in include or ('subtasks', 'attachments'):
                fields['_copy'][f'copy_{i}'] = '1'

        return self.create(**fields)

    def _prepare_create_request(self, request):
        request = super()._prepare_create_request(request)
        request.update(request[self.container].pop('_copy', {}))
        return request


class FileManager(ResourceManager):
    def _process_create_response(self, request, response):
        if response is True:
            response = {self.container: {'id': int(request[self.container]['token'].split('.')[0])}}

        return super()._process_create_response(request, response)


class WikiPageManager(ResourceManager):
    def _process_create_response(self, request, response):
        if response is True:
            raise exceptions.ValidationError('Resource already exists')  # issue #182

        return super()._process_create_response(request, response)


class UserManager(ResourceManager):
    @staticmethod
    def _check_custom_url(path):
        if path.endswith('/me.json'):
            path = '/my/account.json'

        return path

    def _construct_get_url(self, path):
        return super()._construct_get_url(self._check_custom_url(path))

    def all(self, **params):
        resourceset = super().all(**params)

        if self.redmine.ver is not None:  # https://www.redmine.org/issues/32090#note-6
            if self.redmine.ver >= (5, 1, 2):
                resourceset.manager.url = f'{resourceset.manager.url}*'
            elif self.redmine.ver in ((5, 1, 0), (5, 1, 1)):
                resourceset.manager.url = (f'{resourceset.manager.url[:-7]}f[]=status_id&'
                                           f'op[status_id]==&v[status_id][]=1&v[status_id][]=2&v[status_id][]=3')

        return resourceset

    def _prepare_create_request(self, request):
        request = super()._prepare_create_request(request)
        request['send_information'] = request[self.container].pop('send_information', False)
        return request

    def _construct_update_url(self, path):
        return super()._construct_update_url(self._check_custom_url(path))

    def _prepare_update_request(self, request):
        request = super()._prepare_update_request(request)
        request['send_information'] = request[self.resource_class.container_update].pop('send_information', False)
        return request


class NewsManager(ResourceManager):
    def _process_create_response(self, request, response):
        if response is True:
            response = {self.container: self.redmine.news.filter(**self.params)[0].raw()}

        return super()._process_create_response(request, response)