projects
/
python_utils.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
More cleanup, yey!
[python_utils.git]
/
logical_search.py
diff --git
a/logical_search.py
b/logical_search.py
index 4295aa0892fd0a67e8af778aa92f92a30d58e436..b55e68901501ad2e68a7f45df034866072c3f61b 100644
(file)
--- a/
logical_search.py
+++ b/
logical_search.py
@@
-1,5
+1,9
@@
#!/usr/bin/env python3
#!/usr/bin/env python3
+"""This is a module concerned with the creation of and searching of a
+corpus of documents. The corpus is held in memory for fast
+searching."""
+
from __future__ import annotations
import enum
from __future__ import annotations
import enum
@@
-12,6
+16,7
@@
class ParseError(Exception):
"""An error encountered while parsing a logical search expression."""
def __init__(self, message: str):
"""An error encountered while parsing a logical search expression."""
def __init__(self, message: str):
+ super().__init__()
self.message = message
self.message = message
@@
-168,9
+173,7
@@
class Corpus(object):
def invert_docid_set(self, original: Set[str]) -> Set[str]:
"""Invert a set of docids."""
def invert_docid_set(self, original: Set[str]) -> Set[str]:
"""Invert a set of docids."""
- return set(
- [docid for docid in self.documents_by_docid.keys() if docid not in original]
- )
+ return set([docid for docid in self.documents_by_docid.keys() if docid not in original])
def get_doc(self, docid: str) -> Optional[Document]:
"""Given a docid, retrieve the previously added Document."""
def get_doc(self, docid: str) -> Optional[Document]:
"""Given a docid, retrieve the previously added Document."""
@@
-250,9
+253,7
@@
class Corpus(object):
operation = Operation.from_token(token)
operand_count = operation.num_operands()
if len(node_stack) < operand_count:
operation = Operation.from_token(token)
operand_count = operation.num_operands()
if len(node_stack) < operand_count:
- raise ParseError(
- f"Incorrect number of operations for {operation}"
- )
+ raise ParseError(f"Incorrect number of operations for {operation}")
for _ in range(operation.num_operands()):
args.append(node_stack.pop())
node = Node(corpus, operation, args)
for _ in range(operation.num_operands()):
args.append(node_stack.pop())
node = Node(corpus, operation, args)
@@
-342,9
+343,7
@@
class Node(object):
try:
key, value = tag.split(":")
except ValueError as v:
try:
key, value = tag.split(":")
except ValueError as v:
- raise ParseError(
- f'Invalid key:value syntax at "{tag}"'
- ) from v
+ raise ParseError(f'Invalid key:value syntax at "{tag}"') from v
if value == "*":
r = self.corpus.get_docids_with_property(key)
else:
if value == "*":
r = self.corpus.get_docids_with_property(key)
else: