You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

66 lines
1.6 KiB

from panflute import Doc
class Context:
def __init__(self, doc: Doc, path: str, parent: 'Context'=None):
self.parent = parent
self._commands = {}
self.doc = doc
self.path = path
if self.get_metadata("flags", immediate=True) is None:
self.set_metadata("flags", {})
def get_command(self, command: str):
if command in self._commands:
return self._commands[command]
elif self.parent:
return self.parent.get_command(command)
else:
return None
def set_command(self, command: str, val):
self._commands[command] = val
def unset_command(self, command: str):
del self._commands[command]
def is_flag_set(self, flag: str):
if self.get_metadata("flags."+flag):
if self.get_metadata("flags."+flag):
return True
else:
return False
elif self.parent:
return self.parent.is_flag_set(flag)
else:
return False
def set_flag(self, flag: str, val: bool):
self.set_metadata("flags."+flag, val)
def unset_flag(self, flag):
self.unset_metadata("flags."+flag)
def get_metadata(self, key, simple=True, immediate=False):
value = self.doc.get_metadata(key, None, simple)
if value is not None:
return value
elif self.parent and not immediate:
return self.parent.get_metadata(key)
else:
return None
def set_metadata(self, key, value):
meta = self.doc.metadata
key = key.split(".")
for k in key[:-1]:
meta = meta[k]
meta[key[-1]] = value
def unset_metadata(self, key):
meta = self.doc.metadata
key = key.split(".")
for k in key[:-1]:
meta = meta[k]
del meta.content[key[-1]] # A hack because MetaMap doesn't have a __delitem__