'''
WESMERE_FOOTER = '''\
- ')
def abbrev(name):
abbrev = name[0]
word_separators = [" ", "_", "+", "(", ")"]
for i in range(1, len(name)):
if name[i] in ["+", "(", ")"] or \
name[i - 1] in word_separators and \
name[i] not in word_separators:
abbrev += name[i]
return abbrev
def add_menu(menuid, name, classes='', url='', is_table_container=False, container_classes=''):
"""
Writes the initial portion of a sidebar item, including the item's
label, the start tag for the popup menu container, and the label
for said menu.
If is_table_container=True, the popup menu prolog is suitable for
a table (currently used for the Language menu). This will be
removed one day, hopefully. (TODO)
The url parameter allows setting a destination URL for the menu
header, which will be a link if it is set to a non-empty string.
"""
html_name = cleantext(name)
html_classes = " ".join((cleantext(classes), "popuptrigger"))
write('
- ')
write('' + html_name + '')
if menuid:
write('\n')
write('');
def pic(self, u, x, recursion=0):
if recursion >= 4:
error_message(
"Warning: Cannot find image for unit %s(%s).\n" % (
u.get_text_val("id"), x.name.decode("utf8")))
return None, None
image = self.wesnoth.get_unit_value(x, "image")
portrait = self.wesnoth.get_unit_value(x, "profile")
if not portrait:
bu = self.wesnoth.get_base_unit(u)
if bu:
portrait = self.wesnoth.get_unit_value(bu, "profile")
if not image:
if x.name == b"female":
baseunit = self.wesnoth.get_base_unit(u)
if baseunit:
female = baseunit.get_all(tag="female")
return self.pic(u, female[0], recursion=recursion + 1)
else:
return self.pic(u, u, recursion=recursion + 1)
error_message("Warning: Missing image for unit %s(%s).\n" %
(u.get_text_val("id"), x.name.decode("utf8")))
return None, None
icpic = image_collector.add_image_check(self.addon, image)
if not icpic.ipath:
error_message("Warning: No picture %s for unit %s.\n" %
(image, u.get_text_val("id")))
picname = icpic.id_name
image = os.path.join(PICS_LOCATION, picname)
if portrait:
picname = image_collector.add_image(self.addon,
portrait,
no_tc=True,
check_transparent=True)
portrait = os.path.join(PICS_LOCATION, picname)
return image, portrait
def get_abilities(self, u):
anames = []
already = {}
for abilities in u.get_all(tag="abilities"):
try:
c = abilities.get_all()
except AttributeError:
c = []
for ability in c:
try:
id = ability.get_text_val("id")
except AttributeError as e:
error_message("Error: Ignoring ability " + ability.debug())
continue
if id in already:
continue
already[id] = True
name = T(ability, "name")
if not name:
name = ability.name.decode("utf8")
# Only add abilities with a label, since those that lack one
# are normally hidden in the game and used to implement more
# complex ones.
if name:
anames.append(name)
return anames
def get_recursive_attacks(self, this_unit):
def copy_attributes(copy_from, copy_to):
for c in copy_from.data:
if isinstance(c, wmlparser3.AttributeNode):
copy_to.data.append(c)
# Use attacks of base_units as base, if we have one.
base_unit = self.wesnoth.get_base_unit(this_unit)
attacks = []
if base_unit:
attacks = copy.deepcopy(self.get_recursive_attacks(base_unit))
base_attacks_count = len(attacks)
for i, attack in enumerate(this_unit.get_all(tag="attack")):
# Attack merging is order based.
if i < base_attacks_count:
copy_attributes(attack, attacks[i])
else:
attacks.append(attack)
return attacks
def write_units(self):
def write(line):
self.output.write(line)
def _(msgid, textdomain="wesnoth"):
return self.translate(msgid, textdomain)
rows = self.unitgrid
write('
\n
\n') def write_units_tree(self, grouper, title, add_parents): html_title = cleantext(title) self.output.write(website_header(title=html_title, classes=['wmlunits-tree'])) n = self.analyze_units(grouper, add_parents) self.write_navbar("units_tree") self.output.write('') for i in range(6): write(' ') pic = image_collector.add_image("general", "../../../images/misc/leader-crown.png", no_tc=True) crownimage = cleanurl(os.path.join(PICS_LOCATION, pic)) ms = None for row in range(len(rows)): write('' % i) write(' \n') for column in range(6): hspan, vspan, un = rows[row][column] if vspan: attributes = "" if hspan == 1 and vspan == 1: pass elif hspan == 1: attributes += ' rowspan="%d"' % vspan elif vspan == 1: attributes += ' colspan="%d"' % hspan if un and isinstance(un, helpers.GroupNode): # Find the current multiplayer side so we can show the # little crowns.. ms = None if self.is_era: try: eid, fid = un.data era = self.wesnoth.era_lookup[eid] if fid: ms = era.faction_lookup[fid] except TypeError: pass racename = un.name # TODO: we need to use a unique race id instead of a potentially duplicate # name for the header id and link target! attributes += ' id="%s" class="raceheader"' % cleantext(racename) write(' \n') write('L%s%s' % (level, crown)) write('%s
' % (link, uid, name)) write('') image, portrait = self.pic(u, u) image = cleanurl(image) write('' % (link, uid)) if crown == " ♚": write('\n') write('' % image) write('') else: write('' % crownimage) write('
') write('') write('\n') else: write('
') # Write info about abilities. anames = self.get_abilities(u) if anames: write('\n') attributes = ( (_("Cost: "), cost), (_("HP: "), hp), (_("XP: "), xp), (_("MP: "), mp), ) for attr_label, attr_value in attributes: write(' %s %s ') write(cleantext(", ".join(anames))) write('') # Write info about attacks. attacks = self.get_recursive_attacks(u) if attacks: write('\n') first_attack = True for attack in attacks: if not first_attack: write('') write('
') first_attack = False n = T(attack, "number") x = T(attack, "damage") x = "%s %s %s " % (cleantext(x, quote=False), HTML_ENTITY_MULTIPLICATION_SIGN, cleantext(n, quote=False)) write(x) r = T(attack, "range") t = T(attack, "type") write(cleantext("%s (%s)" % (_(r), _(t)), quote=False)) s = [] specials = attack.get_all(tag="specials") if specials: for special in specials[0].get_all(tag=""): sname = T(special, "name") if sname: s.append(sname) s = ", ".join(s) if s: write(" (%s)" % cleantext(s, quote=False)) write('') write(' ') self.output.write('') self.output.write(website_footer()) return n def write_unit_report(self, output, unit): def write(line): self.output.write(line) def _(msgid, textdomain="wesnoth"): return self.translate(msgid, textdomain) def find_attr(what, key): if unit.movetype: mtx = unit.movetype.get_all(tag=what) mty = None if mtx: mty = mtx[0].get_text_val(key) x = unit.get_all(tag=what) y = None if x: y = x[0].get_text_val(key, translation=self.translation.translate) if y: return True, y if unit.movetype and mty is not None: return False, mty return False, "-" def uval(name): return self.wesnoth.get_unit_value(unit, name, translation=self.translation.translate) def clean_uval(name): return cleantext(uval(name)) # Write unit name, picture and description. uid = unit.get_text_val("id") uname = uval("name") display_name = cleantext(uname) self.output = output write(website_header(title=display_name, classes=['wmlunits-unit'])) self.write_navbar("unit_report") self.output.write('%s
' % html_title) self.write_units() self.output.write(HTML_CLEAR_FLOATS) self.output.write(build_timestamp()) self.output.write('') female = unit.get_all(tag="female") if female: fname = T(female[0], "name") if fname and fname != uname: display_name += " / " + cleantext(fname) write('') # main self.output.write(website_footer()) def generate_campaign_report(addon, isocode, campaign, wesnoth): if campaign: cid = campaign.get_text_val("id") else: cid = "mainline" if not cid: cid = addon + "_" + campaign.get_text_val("define") print(("campaign " + addon + " " + cid + " " + isocode)) path = os.path.join(options.output, addon, isocode) if not os.path.isdir(path): os.mkdir(path) output = MyFile(os.path.join(path, "%s.html" % cid), "w") html = HTMLOutput(isocode, output, addon, cid, False, wesnoth) html.target = "%s.html" % cid grouper = GroupByRace(wesnoth, cid) if campaign: title = campaign.get_text_val("name", translation=html.translate) else: title = html.translate("Units", "wesnoth-help") if not title: title = cid n = html.write_units_tree(grouper, title, True) output.close() return n def generate_era_report(addon, isocode, era, wesnoth): eid = era.get_text_val("id") print("era " + addon + " " + eid + " " + isocode) path = os.path.join(options.output, addon, isocode) if not os.path.isdir(path): os.mkdir(path) output = MyFile(os.path.join(path, "%s.html" % eid), "w") html = HTMLOutput(isocode, output, addon, eid, True, wesnoth) html.target = "%s.html" % eid grouper = GroupByFaction(wesnoth, eid) ename = era.get_text_val("name", translation=html.translate) n = html.write_units_tree(grouper, ename, False) output.close() return n def generate_single_unit_reports(addon, isocode, wesnoth): path = os.path.join(options.output, addon, isocode) if not os.path.isdir(path): os.mkdir(path) html = HTMLOutput(isocode, None, addon, "units", False, wesnoth) grouper = GroupByNothing() html.analyze_units(grouper, True) for uid, unit in list(wesnoth.unit_lookup.items()): if unit.hidden: continue if "mainline" in unit.campaigns and addon != "mainline": continue try: htmlname = "%s.html" % uid filename = os.path.join(path, htmlname) # We probably can come up with something better. if os.path.exists(filename): age = time.time() - os.path.getmtime(filename) # was modified in the last 12 hours - we should be ok if age < 3600 * 12: continue except (UnicodeDecodeError, UnicodeEncodeError) as e: traceback.print_exc() error_message("Unicode problem: " + repr(path) + " + " + repr(uid) + "\n") error_message(str(e) + "\n") continue output = MyFile(filename, "w") html.target = "%s.html" % uid html.write_unit_report(output, unit) output.close() def html_postprocess_file(filename, isocode, batchlist): print("postprocessing " + repr(filename)) chtml = "" ehtml = "" cids = [[], []] for addon in batchlist: for campaign in addon.get("campaigns", []): if campaign["units"] == "?" or campaign["units"] <= 0: continue lang = isocode if addon["name"] == "mainline" else "en_US" c = addon["name"], campaign["id"], campaign["translations"].get( lang, campaign["name"]), lang if addon["name"] == "mainline": cids[0].append(c) else: cids[1].append(c) for i in range(2): campaigns = cids[i] campaigns.sort(key=lambda x: "A" if x[1] == "mainline" else "B" + x[2]) for campaign in campaigns: addon, cname, campname, lang = campaign url = cleanurl("../../%s/%s/%s.html" % (addon, lang, cname)) chtml += '\n') write('\n') write('\n') write('\n') write('%s
\n' % display_name) write('') if female: mimage, portrait = self.pic(unit, unit) fimage, fportrait = self.pic(unit, female[0]) if not fimage: fimage = mimage if not fportrait: fportrait = portrait write('\n') description = clean_uval("description") # TODO: what is unit_description? if not description: description = clean_uval("unit_description") if not description: description = HTML_ENTITY_HORIZONTAL_BAR write('%s
\n' % re.sub('\n', '\n
', description)) write('\n') for si in range(2): if si and not female: break if si: sportrait = fportrait simage = fimage else: simage = image sportrait = portrait write('\n') write('') if portrait: write('') write('' % cleanurl(simage)) write('\n\n') self.output.write(HTML_CLEAR_FLOATS) self.output.write(build_timestamp()) write('\n') write('') write('Information
\n') write('\n') # Advances-from list write('
\n') # Write info about attacks. attacks = self.get_recursive_attacks(unit) if attacks: write('%s ' % cleantext(_("Advances from: "), quote=False)) have_advances = False for pid in self.forest.get_parents(uid): punit = self.wesnoth.unit_lookup[pid] if "mainline" in unit.campaigns and "mainline" not in punit.campaigns: continue addon = "mainline" if "mainline" in unit.campaigns else self.addon link = cleanurl("../../%s/%s/%s.html" % (addon, self.isocode, pid)) name = self.wesnoth.get_unit_value(punit, "name", translation=self.translation.translate) if have_advances: write(', ') write('%s' % (link, cleantext(name, quote=False))) have_advances = True if not have_advances: write(HTML_ENTITY_FIGURE_DASH) write(' %s ' % cleantext(_("Advances to: "), quote=False)) have_advances = False for cid in self.forest.get_children(uid): try: cunit = self.wesnoth.unit_lookup[cid] addon = "mainline" if "mainline" in cunit.campaigns else self.addon link = cleanurl("../../%s/%s/%s.html" % (addon, self.isocode, cid)) if "mainline" in unit.campaigns and "mainline" not in cunit.campaigns: continue name = self.wesnoth.get_unit_value(cunit, "name", translation=self.translation.translate) except KeyError: error_message("Warning: Unit %s not found.\n" % cid) name = cid if "mainline" in unit.campaigns: continue link = cleanurl(self.target) if have_advances: write(', ') write('%s' % (link, cleantext(name, quote=False))) have_advances = True if not have_advances: write(HTML_ENTITY_FIGURE_DASH) write(' %s %s \n') write(' \n') write('%s ' % cleantext(_("Abilities: "), quote=False)) if len(anames): write('' + cleantext(', '.join(anames), quote=False) + ' ') else: write('' + HTML_ENTITY_FIGURE_DASH + ' ') write('%s (damage %s count)
\n' % (cleantext(_("unit help^Attacks"), quote=False), HTML_ENTITY_MULTIPLICATION_SIGN)) write('\n') write('
\n') # Write info about resistances. write('') aid = attack.get_text_val("name") aname = T(attack, "description") icon = attack.get_text_val("icon") if not icon: icon = "attacks/%s.png" % aid image_add = image_collector.add_image_check(self.addon, icon, no_tc=True) if not image_add.ipath: error_message("Error: No attack icon '%s' found for '%s'.\n" % ( icon, uid)) icon = os.path.join(PICS_LOCATION, "unit$elves-wood$shaman.png") else: icon = os.path.join(PICS_LOCATION, image_add.id_name) write(' ') write('%s' % cleantext(aname, quote=False)) r = T(attack, "range") write(' ' % cleantext(_(r), quote=False)) n = attack.get_text_val("number") x = attack.get_text_val("damage") x = '%s %s %s' % (cleantext(x, quote=False), HTML_ENTITY_MULTIPLICATION_SIGN, cleantext(n, quote=False)) write('
%s%s' % x) t = T(attack, "type") write(' ' % cleantext(_(t), quote=False)) s = [] specials = attack.get_all(tag="specials") if specials: for special in specials[0].get_all(tag=""): sname = T(special, "name") if sname: s.append(cleantext(sname, quote=False)) else: error_message("Warning: Weapon special %s has no name for %s.\n" % (special.name.decode("utf8"), uid)) write('
%s%s ' % '
'.join(s)) write('%s
\n' % _("Resistances: ").strip(" :")) write('\n') write('
\n') # end left column write('\n') else: write(' \n') row += 1 write('') picname = image_collector.add_image(self.addon, ricon, no_tc=True) icon = os.path.join(PICS_LOCATION, picname) write(' %s %s \n' % (cleantext(_(rid), quote=False), ' '.join(resist_classes), resist_str)) if row % 2 == 1: write('') # Write info about movement costs and terrain defense. write('\n') write('' + cleantext(_("Terrain"), quote=False) + '
\n') write('\n') write('
\n') write('%s %s %s \n') picname = image_collector.add_image(self.addon, "terrain/" + ticon + ".png", no_tc=True) icon = os.path.join(PICS_LOCATION, picname) write(' \n') write('%s %s %s \n' % (cleantext(tname, quote=False), ' '.join(classes_cost), move_cost, ' '.join(classes_defense), defense)) write(' - %s \n' % ( cleantext(campname), url, cleantext(campname, quote=False)) if i == 0 and cids[1]: chtml += '
- '
eids = [[], []]
for addon in batchlist:
for era in addon.get("eras", []):
if era["units"] == "?" or era["units"] <= 0:
continue
lang = isocode if addon["name"] == "mainline" else "en_US"
e = addon["name"], era["id"], era["translations"].get(
lang, era["name"]), lang
if addon["name"] == "mainline":
eids[0].append(e)
else:
eids[1].append(e)
for i in range(2):
eras = eids[i]
eras.sort(key=lambda x: x[2])
for era in eras:
addon, eid, eraname, lang = era
url = cleanurl("../../%s/%s/%s.html" % (addon, lang, eid))
ehtml += '
- %s \n' % ( cleantext(eraname), url, cleantext(eraname, quote=False)) if i == 0 and eids[1]: ehtml += '
- '
f = open(filename, "r+b")
html = f.read().decode("utf8")
html = html.replace(PRE_PLACEHOLDER_CAMPAIGNS, chtml)
html = html.replace(PRE_PLACEHOLDER_ERAS, ehtml)
f.seek(0)
f.write(html.encode("utf8"))
f.close()
def html_postprocess_all(batchlist):
for isocode, filename in all_written_html_files:
html_postprocess_file(filename, isocode, batchlist)
def write_index(out_path):
output = MyFile(os.path.join(out_path, "index.html"), "w")
output.write("""
Redirecting to Wesnoth units database...
""")