mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
38be6d05d0
* Convert all Python2 to Python3 syntax. * Removes oauth2lib dependency * Replace mockredis with fakeredis * byte/str conversions * Removes nonexisting __nonzero__ in Python3 * Python3 Dockerfile and related * [PROJQUAY-98] Replace resumablehashlib with rehash * PROJQUAY-123 - replace gpgme with python3-gpg * [PROJQUAY-135] Fix unhashable class error * Update external dependencies for Python 3 - Move github.com/app-registry/appr to github.com/quay/appr - github.com/coderanger/supervisor-stdout - github.com/DevTable/container-cloud-config - Update to latest mockldap with changes applied from coreos/mockldap - Update dependencies in requirements.txt and requirements-dev.txt * Default FLOAT_REPR function to str in json encoder and removes keyword assignment True, False, and str were not keywords in Python2... * [PROJQUAY-165] Replace package `bencode` with `bencode.py` - Bencode is not compatible with Python 3.x and is no longer maintained. Bencode.py appears to be a drop-in replacement/fork that is compatible with Python 3. * Make sure monkey.patch is called before anything else ( * Removes anunidecode dependency and replaces it with text_unidecode * Base64 encode/decode pickle dumps/loads when storing value in DB Base64 encodes/decodes the serialized values when storing them in the DB. Also make sure to return a Python3 string instead of a Bytes when coercing for db, otherwise, Postgres' TEXT field will convert it into a hex representation when storing the value. * Implement __hash__ on Digest class In Python 3, if a class defines __eq__() but not __hash__(), its instances will not be usable as items in hashable collections (e.g sets). * Remove basestring check * Fix expected message in credentials tests * Fix usage of Cryptography.Fernet for Python3 (#219) - Specifically, this addresses the issue where Byte<->String conversions weren't being applied correctly. * Fix utils - tar+stream layer format utils - filelike util * Fix storage tests * Fix endpoint tests * Fix workers tests * Fix docker's empty layer bytes * Fix registry tests * Appr * Enable CI for Python 3.6 * Skip buildman tests Skip buildman tests while it's being rewritten to allow ci to pass. * Install swig for CI * Update expected exception type in redis validation test * Fix gpg signing calls Fix gpg calls for updated gpg wrapper, and add signing tests. * Convert / to // for Python3 integer division * WIP: Update buildman to use asyncio instead of trollius. This dependency is considered deprecated/abandoned and was only used as an implementation/backport of asyncio on Python 2.x This is a work in progress, and is included in the PR just to get the rest of the tests passing. The builder is actually being rewritten. * Target Python 3.8 * Removes unused files - Removes unused files that were added accidentally while rebasing - Small fixes/cleanup - TODO tasks comments * Add TODO to verify rehash backward compat with resumablehashlib * Revert "[PROJQUAY-135] Fix unhashable class error" and implements __hash__ instead. This reverts commit 735e38e3c1d072bf50ea864bc7e119a55d3a8976. Instead, defines __hash__ for encryped fields class, using the parent field's implementation. * Remove some unused files ad imports Co-authored-by: Kenny Lee Sin Cheong <kenny.lee@redhat.com> Co-authored-by: Tom McKay <thomasmckay@redhat.com>
109 lines
2.5 KiB
Python
109 lines
2.5 KiB
Python
import datetime
|
|
import json
|
|
import base64
|
|
|
|
from contextlib import contextmanager
|
|
from data import model
|
|
|
|
from flask import g
|
|
from flask_principal import Identity
|
|
|
|
CSRF_TOKEN_KEY = "_csrf_token"
|
|
|
|
|
|
@contextmanager
|
|
def client_with_identity(auth_username, client):
|
|
with client.session_transaction() as sess:
|
|
if auth_username and auth_username is not None:
|
|
loaded = model.user.get_user(auth_username)
|
|
sess["user_id"] = loaded.uuid
|
|
sess["login_time"] = datetime.datetime.now()
|
|
else:
|
|
sess["user_id"] = "anonymous"
|
|
|
|
yield client
|
|
|
|
with client.session_transaction() as sess:
|
|
sess["user_id"] = None
|
|
sess["login_time"] = None
|
|
sess[CSRF_TOKEN_KEY] = None
|
|
|
|
|
|
@contextmanager
|
|
def toggle_feature(name, enabled):
|
|
"""
|
|
Context manager which temporarily toggles a feature.
|
|
"""
|
|
import features
|
|
|
|
previous_value = getattr(features, name)
|
|
setattr(features, name, enabled)
|
|
yield
|
|
setattr(features, name, previous_value)
|
|
|
|
|
|
def add_csrf_param(client, params):
|
|
"""
|
|
Returns a params dict with the CSRF parameter added.
|
|
"""
|
|
params = params or {}
|
|
|
|
with client.session_transaction() as sess:
|
|
params[CSRF_TOKEN_KEY] = "sometoken"
|
|
sess[CSRF_TOKEN_KEY] = "sometoken"
|
|
|
|
return params
|
|
|
|
|
|
def gen_basic_auth(username, password):
|
|
"""
|
|
Generates a basic auth header.
|
|
"""
|
|
encoded_username = username.encode("utf-8")
|
|
encoded_password = password.encode("utf-8")
|
|
return "Basic " + base64.b64encode(b"%s:%s" % (encoded_username, encoded_password)).decode(
|
|
"ascii"
|
|
)
|
|
|
|
|
|
def conduct_call(
|
|
client,
|
|
resource,
|
|
url_for,
|
|
method,
|
|
params,
|
|
body=None,
|
|
expected_code=200,
|
|
headers=None,
|
|
raw_body=None,
|
|
):
|
|
"""
|
|
Conducts a call to a Flask endpoint.
|
|
"""
|
|
params = add_csrf_param(client, params)
|
|
|
|
final_url = url_for(resource, **params)
|
|
|
|
headers = headers or {}
|
|
headers.update({"Content-Type": "application/json"})
|
|
|
|
if body is not None:
|
|
body = json.dumps(body)
|
|
|
|
if raw_body is not None:
|
|
body = raw_body
|
|
|
|
# Required for anonymous calls to not exception.
|
|
g.identity = Identity(None, "none")
|
|
|
|
rv = client.open(final_url, method=method, data=body, headers=headers)
|
|
msg = "%s %s: got %s expected: %s | %s" % (
|
|
method,
|
|
final_url,
|
|
rv.status_code,
|
|
expected_code,
|
|
rv.data,
|
|
)
|
|
assert rv.status_code == expected_code, msg
|
|
return rv
|