summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorRené 'Necoro' Neumann <necoro@necoro.net>2017-03-11 13:46:59 +0100
committerRené 'Necoro' Neumann <necoro@necoro.net>2017-03-11 13:46:59 +0100
commitef269be757593035b3f936de3a81426a86ed2557 (patch)
tree6938b9a32a898aa9ec3cafaaee270f753abe04e8
parente13ba704bcbbc97b5660e5465db714bf6dbbaad0 (diff)
downloadarchivist-ef269be757593035b3f936de3a81426a86ed2557.tar.gz
archivist-ef269be757593035b3f936de3a81426a86ed2557.tar.bz2
archivist-ef269be757593035b3f936de3a81426a86ed2557.zip
Remove peewee import to allow import on global level
-rw-r--r--archivist/cli.py2
-rw-r--r--archivist/prefixes.py6
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