-
Notifications
You must be signed in to change notification settings - Fork 1
/
lputils.py
78 lines (66 loc) · 2.78 KB
/
lputils.py
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
# This file is part of concourse-lp-livefs-resource
#
# Copyright 2023 Canonical Ltd.
#
# This program is free software: you can redistribute it and/or modify it under
# the terms of the GNU Lesser General Public License version 3, as published by
# the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranties of MERCHANTABILITY,
# SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from typing import Tuple
from launchpadlib.credentials import (
Credentials,
CredentialStore,
AccessToken,
RequestTokenAuthorizationEngine,
)
from launchpadlib.launchpad import Launchpad, Consumer
from requests.auth import AuthBase
from requests_oauthlib import OAuth1
class DummyCredentialStore(CredentialStore):
"""This class provides oauth token and secret store"""
def __init__(self, consumer_name, oauth_token, oauth_token_secret):
super(DummyCredentialStore, self).__init__(credential_save_failed=None)
self.credentials = Credentials()
self.credentials.consumer = Consumer(consumer_name)
self.credentials.access_token = AccessToken.from_params(
{"oauth_token": oauth_token, "oauth_token_secret": oauth_token_secret}
)
def do_load(self, unique_key):
return self.credentials
class DummyAuthorizationEngine(RequestTokenAuthorizationEngine):
"""This stub class prevents launchpadlib from nulling out consumer_name
in its demented campaign to force the use of desktop integration.
This was called ShutUpAndTakeMyTokenAuthorizationEngine"""
def __init__(
self,
service_root,
consumer_name,
application_name=None,
allow_access_levels=None,
):
super(DummyAuthorizationEngine, self).__init__(
service_root, application_name, consumer_name, allow_access_levels
)
def login_with_oauth(
consumer_name, oauth_token, oauth_token_secret, service_root="production"
) -> Tuple[Launchpad, AuthBase]:
"""This function provides launchpadlib Launchpad instance with existing
OAuth token and secret passed from elsewhere."""
credstore = DummyCredentialStore(consumer_name, oauth_token, oauth_token_secret)
authengine = DummyAuthorizationEngine(service_root, consumer_name)
lp = Launchpad.login_with(
consumer_name,
service_root,
version="devel",
credential_store=credstore,
authorization_engine=authengine,
)
auth = OAuth1(consumer_name, "", oauth_token, oauth_token_secret)
return lp, auth