mirror of
https://github.com/quay/quay.git
synced 2025-04-18 10:44:06 +03:00
a839a78eb5
Adds ACCOUNT_RECOVERY_MODE to allow Quay to run with some core features disabled. When this is set, the instance should only be used in order by existing users who hasn't linked their account to an external login service, after database authentication has been disabled.
83 lines
2.5 KiB
Python
83 lines
2.5 KiB
Python
import os
|
|
import logging.config
|
|
import time
|
|
import argparse
|
|
|
|
import features
|
|
|
|
from app import app, repo_mirror_api
|
|
from workers.worker import Worker
|
|
from workers.repomirrorworker import process_mirrors
|
|
from util.repomirror.validator import RepoMirrorConfigValidator
|
|
from util.repomirror.skopeomirror import SkopeoMirror
|
|
from util.log import logfile_path
|
|
from workers.gunicorn_worker import GunicornWorker
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
DEFAULT_MIRROR_INTERVAL = 30
|
|
|
|
|
|
class RepoMirrorWorker(Worker):
|
|
def __init__(self):
|
|
super(RepoMirrorWorker, self).__init__()
|
|
RepoMirrorConfigValidator(app.config.get("FEATURE_REPO_MIRROR", False)).valid()
|
|
|
|
self._mirrorer = SkopeoMirror()
|
|
self._next_token = None
|
|
|
|
interval = app.config.get("REPO_MIRROR_INTERVAL", DEFAULT_MIRROR_INTERVAL)
|
|
self.add_operation(self._process_mirrors, interval)
|
|
|
|
def _process_mirrors(self):
|
|
while True:
|
|
assert app.config.get("FEATURE_REPO_MIRROR", False)
|
|
|
|
self._next_token = process_mirrors(self._mirrorer, self._next_token)
|
|
if self._next_token is None:
|
|
break
|
|
|
|
|
|
def create_gunicorn_worker():
|
|
"""
|
|
follows the gunicorn application factory pattern, enabling
|
|
a quay worker to run as a gunicorn worker thread.
|
|
|
|
this is useful when utilizing gunicorn's hot reload in local dev.
|
|
|
|
utilizing this method will enforce a 1:1 quay worker to gunicorn worker ratio.
|
|
"""
|
|
worker = GunicornWorker(__name__, app, RepoMirrorWorker(), features.REPO_MIRROR)
|
|
return worker
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if os.getenv("PYDEV_DEBUG", None):
|
|
import pydevd_pycharm
|
|
|
|
host, port = os.getenv("PYDEV_DEBUG").split(":")
|
|
pydevd_pycharm.settrace(
|
|
host, port=int(port), stdoutToServer=True, stderrToServer=True, suspend=False
|
|
)
|
|
|
|
logging.config.fileConfig(logfile_path(debug=False), disable_existing_loggers=False)
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"mode", metavar="MODE", type=str, nargs="?", default="", choices=["mirror", ""]
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
if app.config.get("ACCOUNT_RECOVERY_MODE", False):
|
|
logger.debug("Quay running in account recovery mode")
|
|
while True:
|
|
time.sleep(100000)
|
|
|
|
if not features.REPO_MIRROR:
|
|
logger.debug("Repository mirror disabled; skipping RepoMirrorWorker")
|
|
while True:
|
|
time.sleep(100000)
|
|
|
|
worker = RepoMirrorWorker()
|
|
worker.start()
|