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

128 lines
3.6 KiB
Python

import yaml
import os
import glob
import re
class Index( object ):
def __init__( self, path ):
self.path = path
self.fields = {}
self.doc = ""
def path_short( self ):
# XXX needs tidy
return "content/pages/index.md"
def open( self ):
self.read()
def write( self ):
outdata = ( '''\
---
%(yaml)s
---
%(md)s
''' % { 'yaml': yaml.dump( self.fields, default_flow_style = False ),
'md': self.doc,
} )
outfile = open( self.path, 'w', encoding = "utf-8" )
outfile.write( outdata )
return outdata
def read( self ):
indata = open( self.path, 'r', encoding = "utf-8" ).read()
( pre, yamldata, docdata ) = re.split( "^---$", indata, flags = re.MULTILINE )
self.fields = yaml.load( yamldata )
self.doc = docdata.strip()
def _toggle( self, fieldname ):
if self.fields.get( fieldname ):
del self.fields[ fieldname ]
else:
self.fields[ fieldname ] = True
return " ".join( [ x for x in [ 'incident', 'degraded', 'maintenance', 'notice', 'ok' ] if self.fields.get( x ) ] )
def toggle_incident( self ):
return self._toggle( 'incident' )
def toggle_degraded( self ):
return self._toggle( 'degraded' )
def toggle_maintenance( self ):
return self._toggle( 'maintenance' )
def toggle_notice( self ):
return self._toggle( 'notice' )
def toggle_ok( self ):
return self._toggle( 'ok' )
def set_flags( self, flags ):
for fieldname in [ 'incident', 'degraded', 'maintenance', 'notice', 'ok' ]:
self.fields[ fieldname ] = fieldname in flags
class Event( Index ):
def __init__( self, path, glob ):
super().__init__( path )
self.glob = glob
def path_short( self ):
# XXX needs tidy
return "content/event/" + os.path.basename( self.path )
def open( self ):
if os.path.exists( self.path ):
self.read()
return False
for path in glob.glob( self.glob ):
if os.path.exists( path ):
self.path = path
self.read()
return False
return True
def set_title( self, title ):
self.fields[ '$title@' ] = title
def set_slug( self, slug ):
self.fields[ '$slug' ] = slug
def set_published( self, published ):
if not self.fields.get( '$date' ):
self.fields[ '$date' ] = {}
if not self.fields[ '$date' ].get( 'published' ):
self.fields[ '$date' ][ 'published' ] = published
return published
def set_started( self, started ):
if not self.fields.get( '$date' ):
self.fields[ '$date' ] = {}
self.fields[ '$date' ][ 'started' ] = started
return started
def set_expected( self, expected ):
if not self.fields.get( '$date' ):
self.fields[ '$date' ] = {}
self.fields[ '$date' ][ 'expected' ] = expected
return expected
def set_finished( self, finished ):
if not self.fields.get( '$date' ):
self.fields[ '$date' ] = {}
self.fields[ '$date' ][ 'finished' ] = finished
return finished
def set_resolved( self, resolved ):
if not self.fields.get( '$date' ):
self.fields[ '$date' ] = {}
self.fields[ '$date' ][ 'resolved' ] = resolved
return resolved
def add_timeline( self, time, line ):
if not self.fields.get( 'timeline' ):
self.fields[ 'timeline' ] = []
self.fields[ 'timeline' ].append( { 'time': time, 'line': line } )