annotate lib/stategraph.py @ 1654:d47832373b34

__nonzero__ is called __bool__ in py3! thanks for nothing, linters
author drewp@bigasterisk.com
date Sun, 19 Sep 2021 13:21:15 -0700
parents 1ff1fbf6b183
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
330
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
1 import datetime, os, inspect
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
2 from dateutil.tz import tzlocal
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
3 from rdflib import Graph, Namespace, Literal
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
4 DCTERMS = Namespace("http://purl.org/dc/terms/")
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
5
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
6 class StateGraph(object):
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
7 """
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
8 helper to create a graph with some of the current state of the world
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
9 """
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
10 def __init__(self, ctx):
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
11 """
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
12 note that we put the time of the __init__ call into the graph
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
13 as its dcterms:modified time.
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
14 """
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
15 self.g = Graph()
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
16 self.ctx = ctx
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
17
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
18 try:
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
19 requestingFile = inspect.stack()[1][1]
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
20 self.g.add((ctx, DCTERMS['creator'],
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
21 Literal(os.path.abspath(requestingFile))))
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
22 except IndexError:
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
23 pass
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
24 self.g.add((ctx, DCTERMS['modified'],
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
25 Literal(datetime.datetime.now(tzlocal()))))
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
26
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
27 def add(self, *args, **kw):
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
28 self.g.add(*args, **kw)
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
29
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
30 def ntLines(self):
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
31 nt = self.g.serialize(format='nt')
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
32 # this canonical order is just for debugging, so the lines are
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
33 # stable when you refresh the file repeatedly
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
34 return sorted(filter(None, nt.splitlines()))
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
35
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
36 def asTrig(self):
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
37 return "%s {\n%s\n}\n" % (self.ctx.n3(), '\n'.join(self.ntLines()))
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
38
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
39 def asJsonLd(self):
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
40 return self.g.serialize(format='json-ld')
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
41
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
42 def asAccepted(self, acceptHeader):
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
43 if acceptHeader == 'application/nquads':
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
44 return 'application/nquads', '\n'.join(
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
45 line.strip().rstrip('.') + '%s .' % self.ctx.n3()
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
46 for line in self.ntLines())
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
47 else:
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
48 return 'application/x-trig', self.asTrig()
1ff1fbf6b183 move stategraph from magma
drewp@bigasterisk.com
parents:
diff changeset
49