From ef269be757593035b3f936de3a81426a86ed2557 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ren=C3=A9=20=27Necoro=27=20Neumann?= Date: Sat, 11 Mar 2017 13:46:59 +0100 Subject: Remove peewee import to allow import on global level --- archivist/cli.py | 2 ++ archivist/prefixes.py | 6 ++++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/archivist/cli.py b/archivist/cli.py index 80b6094..0c74e3b 100644 --- a/archivist/cli.py +++ b/archivist/cli.py @@ -1,5 +1,7 @@ import click +from .prefixes import query_pseudo_prefix, is_pseudo_prefix + CONTEXT_SETTINGS = dict(help_option_names=['-h', '--help']) def enable_debug(): diff --git a/archivist/prefixes.py b/archivist/prefixes.py index d49729a..596feb0 100644 --- a/archivist/prefixes.py +++ b/archivist/prefixes.py @@ -1,5 +1,3 @@ -from peewee import Field - prefixes = {} registered = False @@ -32,6 +30,10 @@ def query_pseudo_prefix(prefix, value): return prefixes[prefix].as_query(value) return None +def is_pseudo_prefix(prefix): + register_prefixes() + return prefix in prefixes + class PseudoPrefix: def __init__(self, name, description, accessor): self.name = name -- cgit v1.2.3