mirror of
https://github.com/quay/quay.git
synced 2026-01-27 18:42:52 +03:00
5f63b3a7bb
* chore: drop deprecated tables and remove unused code * isort imports * migration: check for table existence before drop
34 lines
737 B
Python
34 lines
737 B
Python
from cachetools.func import lru_cache
|
|
|
|
from data.database import LabelSourceType, MediaType
|
|
|
|
|
|
@lru_cache(maxsize=1)
|
|
def get_label_source_types():
|
|
source_type_map = {}
|
|
for kind in LabelSourceType.select():
|
|
source_type_map[kind.id] = kind.name
|
|
source_type_map[kind.name] = kind.id
|
|
|
|
return source_type_map
|
|
|
|
|
|
@lru_cache(maxsize=1)
|
|
def get_media_types():
|
|
media_type_map = {}
|
|
for kind in MediaType.select():
|
|
media_type_map[kind.id] = kind.name
|
|
media_type_map[kind.name] = kind.id
|
|
|
|
return media_type_map
|
|
|
|
|
|
def _get_label_source_type_id(name):
|
|
kinds = get_label_source_types()
|
|
return kinds[name]
|
|
|
|
|
|
def _get_media_type_id(name):
|
|
kinds = get_media_types()
|
|
return kinds[name]
|