# HG changeset patch # User Marcin Cieślak # Date 1431372910 0 # Mon May 11 19:35:10 2015 +0000 # Node ID 7c8e88ed5b67b3c4d5fbb023c597b4c141f77581 # Parent 775cce38c446638d8c636d594dde3d33bcb20935 Reformat: squash newlines diff --git a/plwiki/chemia/comp.py b/plwiki/chemia/comp.py --- a/plwiki/chemia/comp.py +++ b/plwiki/chemia/comp.py @@ -121,32 +121,44 @@ print >>reallybadfile, repr(comp.title) return (compounds, badlist) +def formatgroup(group, ch, elementopen, cntcur, cntcursiz): + if group: + if cntcursiz > 5: + group.insert(0, u"===%s%d===" % (ch.symbol, cntcur)) + else: + if elementopen: + group.insert(0, "") # Additional newline + return "\n".join(group) + else: + return None + + def formatter(elements): out = ["__NOEDITSECTION__", u"{{Spis treści}}"] + group = [] for (ch, ac) in elements.allarticles(): out.append(u"==%s==" % unicode(ch.wikilink)) ac.sort() group = [] - cntcur = 0 + cntcur = 1 cntcursiz = 0 - groupopen = False + elementopen = False for (cnt, a) in ac: if cnt > cntcur: - out.append("\n".join(group)) - group = [] + got = formatgroup(group, ch, elementopen, cntcur, cntcursiz) + if got: + out.append(got) + group = [] + elementopen = True cntcursiz = 0 cntcur = cnt - if groupopen: - groupopen = False - - if cntcursiz > 5 and not groupopen: - out.append(u"===%s%d===" % (ch.symbol, cnt)) - groupopen = True group.append(u"%s" % a.wikilink()) cntcursiz += 1 - out.append("\n".join(group)) - + + got = formatgroup(group, ch, elementopen, cntcur, cntcursiz) + if got: + out.append(got) return out