summaryrefslogtreecommitdiff
path: root/tools
diff options
context:
space:
mode:
authorHaru <haru@dotalux.com>2018-05-01 18:56:28 +0200
committerGitHub <noreply@github.com>2018-05-01 18:56:28 +0200
commit75930315136d7a6b3ce301d09065e3b0386806e0 (patch)
tree539679fec485b593c54dd2ee4096630232638ed4 /tools
parent24360ce305c85b34d814283be9fe7c70a1eff404 (diff)
parent51da42665a00756ecbeae5e264875c312fd1536d (diff)
downloadhercules-75930315136d7a6b3ce301d09065e3b0386806e0.tar.gz
hercules-75930315136d7a6b3ce301d09065e3b0386806e0.tar.bz2
hercules-75930315136d7a6b3ce301d09065e3b0386806e0.tar.xz
hercules-75930315136d7a6b3ce301d09065e3b0386806e0.zip
Merge pull request #2019 from Asheraf/mskillconf
Convert mob_skill_db into libconf format
Diffstat (limited to 'tools')
-rw-r--r--tools/mobskilldbconverter.py264
-rw-r--r--tools/utils/__init__.py0
-rw-r--r--tools/utils/common.py50
-rw-r--r--tools/utils/libconf.py693
4 files changed, 1007 insertions, 0 deletions
diff --git a/tools/mobskilldbconverter.py b/tools/mobskilldbconverter.py
new file mode 100644
index 000000000..310331043
--- /dev/null
+++ b/tools/mobskilldbconverter.py
@@ -0,0 +1,264 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+#
+# This file is part of Hercules.
+# http://herc.ws - http://github.com/HerculesWS/Hercules
+#
+# Copyright (C) 2018 Hercules Dev Team
+# Copyright (C) 2018 Asheraf
+#
+# Hercules is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+import re
+import sys
+import utils.common as Tools
+
+SKILL_STATES = {
+ "any": "MSS_ANY",
+ "idle": "MSS_IDLE",
+ "walk": "MSS_WALK",
+ "loot": "MSS_LOOT",
+ "dead": "MSS_DEAD",
+ "attack": "MSS_BERSERK",
+ "angry": "MSS_ANGRY",
+ "chase": "MSS_RUSH",
+ "follow": "MSS_FOLLOW",
+ "anytarget": "MSS_ANYTARGET"
+}
+SKILL_COND1 = {
+ "always": "MSC_ALWAYS",
+ "myhpltmaxrate": "MSC_MYHPLTMAXRATE",
+ "myhpinrate": "MSC_MYHPINRATE",
+ "friendhpltmaxrate": "MSC_FRIENDHPLTMAXRATE",
+ "friendhpinrate": "MSC_FRIENDHPINRATE",
+ "mystatuson": "MSC_MYSTATUSON",
+ "mystatusoff": "MSC_MYSTATUSOFF",
+ "friendstatuson": "MSC_FRIENDSTATUSON",
+ "friendstatusoff": "MSC_FRIENDSTATUSOFF",
+ "attackpcgt": "MSC_ATTACKPCGT",
+ "attackpcge": "MSC_ATTACKPCGE",
+ "slavelt": "MSC_SLAVELT",
+ "slavele": "MSC_SLAVELE",
+ "closedattacked": "MSC_CLOSEDATTACKED",
+ "longrangeattacked": "MSC_LONGRANGEATTACKED",
+ "skillused": "MSC_SKILLUSED",
+ "afterskill": "MSC_AFTERSKILL",
+ "casttargeted": "MSC_CASTTARGETED",
+ "rudeattacked": "MSC_RUDEATTACKED",
+ "masterhpltmaxrate": "MSC_MASTERHPLTMAXRATE",
+ "masterattacked": "MSC_MASTERATTACKED",
+ "alchemist": "MSC_ALCHEMIST",
+ "onspawn": "MSC_SPAWN"
+}
+SKILL_COND2 = {
+ "anybad": "MSC_ANY",
+ "stone": "SC_STONE",
+ "freeze": "SC_FREEZE",
+ "stun": "SC_STUN",
+ "sleep": "SC_SLEEP",
+ "poison": "SC_POISON",
+ "curse": "SC_CURSE",
+ "silence": "SC_SILENCE",
+ "confusion": "SC_CONFUSION",
+ "blind": "SC_BLIND",
+ "hiding": "SC_HIDING",
+ "sight": "SC_SIGHT"
+}
+SKILL_TARGET = {
+ "target": "MST_TARGET",
+ "randomtarget": "MST_RANDOM",
+ "self": "MST_SELF",
+ "friend": "MST_FRIEND",
+ "master": "MST_MASTER",
+ "around5": "MST_AROUND5",
+ "around6": "MST_AROUND6",
+ "around7": "MST_AROUND7",
+ "around8": "MST_AROUND8",
+ "around1": "MST_AROUND1",
+ "around2": "MST_AROUND2",
+ "around3": "MST_AROUND3",
+ "around4": "MST_AROUND4",
+ "around": "MST_AROUND"
+}
+
+def printHeader():
+ print("""
+mob_skill_db:(
+{
+/**************************************************************************
+ ************* Entry structure ********************************************
+ **************************************************************************
+ <Monster_Constant>: {
+ <Skill_Constant>: {
+ ClearSkills: (boolean, defaults to false) allows cleaning all previous defined skills for the mob.
+ SkillLevel: (int, defaults to 1)
+ SkillState: (int, defaults to 0)
+ SkillTarget: (int, defaults to 0)
+ Rate: (int, defaults to 1)
+ CastTime: (int, defaults to 0)
+ Delay: (int, defaults to 0)
+ Cancelable: (boolean, defaults to false)
+ CastCondition: (int, defaults to 0)
+ ConditionData: (int, defaults to 0)
+ val0: (int, defaults to 0)
+ val1: (int, defaults to 0)
+ val2: (int, defaults to 0)
+ val3: (int, defaults to 0)
+ val4: (int, defaults to 0)
+ Emotion: (int, defaults to 0)
+ ChatMsgID: (int, defaults to 0)
+ }
+ }
+**************************************************************************/""")
+
+def printFooter():
+ print('}\n)\n')
+
+def isValidEntry(line):
+ if re.match('^[0-9]+,.*', line):
+ return True
+ return False
+
+def commaSplit(line):
+ return line.split(',')
+
+def stripLinebreak(line):
+ return line.replace('\r', '').replace('\n', '')
+
+def printInt(key, value):
+ if key in value:
+ if int(value[key]) is not 0:
+ print('\t\t\t{}: {}'.format(key, value[key]))
+
+def printStrToInt(key, value):
+ if value[key] is not '':
+ if int(value[key]) is not 0:
+ print('\t\t\t{}: {}'.format(key, value[key]))
+
+def printBool(key, value):
+ if value[key] == 'yes':
+ print('\t\t\t{}: true'.format(key))
+
+def printClearSkills(key, value):
+ if value[key] == 'clear':
+ print('\t\t\t{}: true'.format(key))
+
+def printSkillState(key, value):
+ if value[key]:
+ print('\t\t\t{}: "{}"'.format(key, SKILL_STATES[value[key]]))
+
+def printSkillTarget(key, value):
+ if value[key]:
+ print('\t\t\t{}: "{}"'.format(key, SKILL_TARGET[value[key]]))
+
+def printCastCondition(key, value):
+ if value[key]:
+ print('\t\t\t{}: "{}"'.format(key, SKILL_COND1[value[key]]))
+
+def printConditionData(key, value):
+ if value[key] in SKILL_COND2:
+ print('\t\t\t{}: "{}"'.format(key, SKILL_COND2[value[key]]))
+ elif value[key] is not '':
+ if int(value[key]) is not 0:
+ print('\t\t\t{}: {}'.format(key, value[key]))
+
+def printEmotion(key, value):
+ if value[key] is not '':
+ print('\t\t\t{}: {}'.format(key, value[key]))
+
+def LoadOldDB(mode, serverpath):
+
+ r = open('{}db/{}/mob_skill_db.txt'.format(serverpath, mode), "r")
+
+ Db = dict()
+ for line in r:
+ if isValidEntry(line) == True:
+ entry = commaSplit(stripLinebreak(line))
+ MonsterId = entry[0]
+ if MonsterId not in Db:
+ Db[MonsterId] = dict()
+ skillidx = len(Db[MonsterId])
+ Db[MonsterId][skillidx] = dict()
+ Db[MonsterId][skillidx]['ClearSkills'] = entry[1]
+ Db[MonsterId][skillidx]['SkillState'] = entry[2]
+ Db[MonsterId][skillidx]['SkillId'] = entry[3]
+ Db[MonsterId][skillidx]['SkillLevel'] = entry[4]
+ Db[MonsterId][skillidx]['Rate'] = entry[5]
+ Db[MonsterId][skillidx]['CastTime'] = entry[6]
+ Db[MonsterId][skillidx]['Delay'] = entry[7]
+ Db[MonsterId][skillidx]['Cancelable'] = entry[8]
+ Db[MonsterId][skillidx]['SkillTarget'] = entry[9]
+ Db[MonsterId][skillidx]['CastCondition'] = entry[10]
+ Db[MonsterId][skillidx]['ConditionData'] = entry[11]
+ for i in range(5):
+ if entry[12 + i] is '':
+ continue
+ try:
+ Db[MonsterId][skillidx]['val{}'.format(i)] = int(entry[12 + i])
+ except:
+ Db[MonsterId][skillidx]['val{}'.format(i)] = int(entry[12 + i], 16)
+ Db[MonsterId][skillidx]['Emotion'] = entry[17]
+ Db[MonsterId][skillidx]['ChatMsgID'] = entry[18]
+ return Db
+
+def ConvertDB(mode, serverpath):
+ db = LoadOldDB(mode, serverpath)
+ MobDB = Tools.LoadDBConsts('mob_db', mode, serverpath)
+ SkillDB = Tools.LoadDBConsts('skill_db', mode, serverpath)
+
+ printHeader()
+ for mobid in sorted(db.iterkeys()):
+ print('\t{}: {{'.format(MobDB[int(mobid)]))
+ for skillidx in sorted(db[mobid].iterkeys()):
+ valid = True
+ if int(db[mobid][skillidx]['SkillId']) not in SkillDB:
+ valid = False
+ print('/*')
+ print('// Can\'t find skill with id {} in skill_db'.format(db[mobid][skillidx]['SkillId']))
+ print('\t\t{}: {{'.format(db[mobid][skillidx]['SkillId']))
+ else:
+ print('\t\t{}: {{'.format(SkillDB[int(db[mobid][skillidx]['SkillId'])]))
+ printClearSkills('ClearSkills', db[mobid][skillidx])
+ printSkillState('SkillState', db[mobid][skillidx])
+ printStrToInt('SkillLevel', db[mobid][skillidx])
+ printStrToInt('Rate', db[mobid][skillidx])
+ printStrToInt('CastTime', db[mobid][skillidx])
+ printStrToInt('Delay', db[mobid][skillidx])
+ printBool('Cancelable', db[mobid][skillidx])
+ printSkillTarget('SkillTarget', db[mobid][skillidx])
+ printCastCondition('CastCondition', db[mobid][skillidx])
+ printConditionData('ConditionData', db[mobid][skillidx])
+ for i in range(5):
+ printInt('val{}'.format(i), db[mobid][skillidx])
+ printEmotion('Emotion', db[mobid][skillidx])
+ printStrToInt('ChatMsgID', db[mobid][skillidx])
+ print('\t\t}')
+ if valid is False:
+ print('*/')
+ print('\t}')
+ printFooter()
+
+if len(sys.argv) != 3:
+ print('Monster Skill db converter from txt to conf format')
+ print('Usage:')
+ print(' mobskilldbconverter.py mode serverpath')
+ print("example:")
+ print(' mobskilldbconverter.py pre-re ../')
+ exit(1)
+
+if sys.argv[1] != 're' and sys.argv[1] != 'pre-re':
+ print('you have entred an invalid server mode')
+ exit(1)
+
+ConvertDB(sys.argv[1], sys.argv[2])
diff --git a/tools/utils/__init__.py b/tools/utils/__init__.py
new file mode 100644
index 000000000..e69de29bb
--- /dev/null
+++ b/tools/utils/__init__.py
diff --git a/tools/utils/common.py b/tools/utils/common.py
new file mode 100644
index 000000000..1a398d544
--- /dev/null
+++ b/tools/utils/common.py
@@ -0,0 +1,50 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+#
+# This file is part of Hercules.
+# http://herc.ws - http://github.com/HerculesWS/Hercules
+#
+# Copyright (C) 2018 Hercules Dev Team
+# Copyright (C) 2018 Asheraf
+#
+# Hercules is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+import io
+import libconf as libconf
+import os.path
+
+def LoadDBConsts(DBname, mode, serverpath):
+ filenames = [serverpath + 'db/{}/{}.conf'.format(mode, DBname)]
+
+ if os.path.isfile(serverpath + 'db/{}2.conf'.format(DBname)):
+ filenames.append(serverpath + 'db/{}2.conf'.format(DBname))
+
+ consts = dict()
+ for filename in filenames:
+ with io.open(filename) as f:
+ config = libconf.load(f)
+ db = config[DBname]
+ if DBname is 'item_db':
+ for i, v in enumerate(db):
+ consts[db[i].Id] = db[i].AegisName
+ elif DBname is 'mob_db':
+ for i, v in enumerate(db):
+ consts[db[i].Id] = db[i].SpriteName
+ elif DBname is 'skill_db':
+ for i, v in enumerate(db):
+ consts[db[i].Id] = db[i].Name
+ else:
+ print('LoadDBConsts: invalid database name {}'.format(DBname))
+ exit(1)
+ return consts
diff --git a/tools/utils/libconf.py b/tools/utils/libconf.py
new file mode 100644
index 000000000..635efd07d
--- /dev/null
+++ b/tools/utils/libconf.py
@@ -0,0 +1,693 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+#
+# Copyright (C) 2018 Hercules Dev Team
+#
+# This library is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+# This file originally licensed under the MIT License
+#
+# Copyright (c) 2016 Christian Aichinger <Greek0@gmx.net>
+# https://github.com/Grk0/python-libconf
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
+# of the Software, and to permit persons to whom the Software is furnished to do
+# so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+from __future__ import absolute_import, division, print_function
+
+import sys
+import os
+import codecs
+import collections
+import io
+import re
+
+# Define an isstr() and isint() that work on both Python2 and Python3.
+# See http://stackoverflow.com/questions/11301138
+try:
+ basestring # attempt to evaluate basestring
+
+ def isstr(s):
+ return isinstance(s, basestring)
+
+ def isint(i):
+ return isinstance(i, (int, long))
+except NameError:
+
+ def isstr(s):
+ return isinstance(s, str)
+
+ def isint(i):
+ return isinstance(i, int)
+
+# Bounds to determine when an "L" suffix should be used during dump().
+SMALL_INT_MIN = -2**31
+SMALL_INT_MAX = 2**31 - 1
+
+ESCAPE_SEQUENCE_RE = re.compile(r'''
+ ( \\x.. # 2-digit hex escapes
+ | \\[\\'"abfnrtv] # Single-character escapes
+ )''', re.UNICODE | re.VERBOSE)
+
+SKIP_RE = re.compile(r'\s+|#.*$|//.*$|/\*(.|\n)*?\*/', re.MULTILINE)
+UNPRINTABLE_CHARACTER_RE = re.compile(r'[\x00-\x1F\x7F]')
+
+
+# load() logic
+##############
+
+def decode_escapes(s):
+ '''Unescape libconfig string literals'''
+ def decode_match(match):
+ return codecs.decode(match.group(0), 'unicode-escape')
+
+ return ESCAPE_SEQUENCE_RE.sub(decode_match, s)
+
+
+class AttrDict(collections.OrderedDict):
+ '''OrderedDict subclass giving access to string keys via attribute access
+
+ This class derives from collections.OrderedDict. Thus, the original
+ order of the config entries in the input stream is maintained.
+ '''
+
+ def __getattr__(self, attr):
+ if attr == '_OrderedDict__root':
+ # Work around Python2's OrderedDict weirdness.
+ raise AttributeError("AttrDict has no attribute %r" % attr)
+ return self.__getitem__(attr)
+
+
+class ConfigParseError(RuntimeError):
+ '''Exception class raised on errors reading the libconfig input'''
+ pass
+
+
+class ConfigSerializeError(TypeError):
+ '''Exception class raised on errors serializing a config object'''
+ pass
+
+
+class Token(object):
+ '''Base class for all tokens produced by the libconf tokenizer'''
+ def __init__(self, type, text, filename, row, column):
+ self.type = type
+ self.text = text
+ self.filename = filename
+ self.row = row
+ self.column = column
+
+ def __str__(self):
+ return "%r in %r, row %d, column %d" % (
+ self.text, self.filename, self.row, self.column)
+
+
+class FltToken(Token):
+ '''Token subclass for floating point values'''
+ def __init__(self, *args, **kwargs):
+ super(FltToken, self).__init__(*args, **kwargs)
+ self.value = float(self.text)
+
+
+class IntToken(Token):
+ '''Token subclass for integral values'''
+ def __init__(self, *args, **kwargs):
+ super(IntToken, self).__init__(*args, **kwargs)
+ self.is_long = self.text.endswith('L')
+ self.is_hex = (self.text[1:2].lower() == 'x')
+ self.value = int(self.text.rstrip('L'), 0)
+
+
+class BoolToken(Token):
+ '''Token subclass for booleans'''
+ def __init__(self, *args, **kwargs):
+ super(BoolToken, self).__init__(*args, **kwargs)
+ self.value = (self.text[0].lower() == 't')
+
+
+class StrToken(Token):
+ '''Token subclass for strings'''
+ def __init__(self, *args, **kwargs):
+ super(StrToken, self).__init__(*args, **kwargs)
+ self.value = decode_escapes(self.text[1:-1])
+
+
+def compile_regexes(token_map):
+ return [(cls, type, re.compile(regex))
+ for cls, type, regex in token_map]
+
+
+class Tokenizer:
+ '''Tokenize an input string
+
+ Typical usage:
+
+ tokens = list(Tokenizer("<memory>").tokenize("""a = 7; b = ();"""))
+
+ The filename argument to the constructor is used only in error messages, no
+ data is loaded from the file. The input data is received as argument to the
+ tokenize function, which yields tokens or throws a ConfigParseError on
+ invalid input.
+
+ Include directives are not supported, they must be handled at a higher
+ level (cf. the TokenStream class).
+ '''
+
+ token_map = compile_regexes([
+ (FltToken, 'float', r'([-+]?(\d+)?\.\d*([eE][-+]?\d+)?)|'
+ r'([-+]?(\d+)(\.\d*)?[eE][-+]?\d+)'),
+ (IntToken, 'hex64', r'0[Xx][0-9A-Fa-f]+(L(L)?)'),
+ (IntToken, 'hex', r'0[Xx][0-9A-Fa-f]+'),
+ (BoolToken, 'boolean', r'(?i)(true|false)\b'),
+ (StrToken, 'string', r'"([^"\\]|\\.)*"'),
+ (StrToken, 'string', r'<"(?<=<")([\S\s]*?)(?=">)">'),
+ (Token, 'name', r'[0-9]*[A-Za-z\*][-A-Za-z0-9_\*]*'),
+ (IntToken, 'integer64', r'[-+]?[0-9]+L(L)?'),
+ (IntToken, 'integer', r'[-+]?[0-9]+'),
+ (Token, '}', r'\}'),
+ (Token, '{', r'\{'),
+ (Token, ')', r'\)'),
+ (Token, '(', r'\('),
+ (Token, ']', r'\]'),
+ (Token, '[', r'\['),
+ (Token, ',', r','),
+ (Token, ';', r';'),
+ (Token, '=', r'='),
+ (Token, ':', r':'),
+ ])
+
+ def __init__(self, filename):
+ self.filename = filename
+ self.row = 1
+ self.column = 1
+
+ def tokenize(self, string):
+ '''Yield tokens from the input string or throw ConfigParseError'''
+ pos = 0
+ while pos < len(string):
+ m = SKIP_RE.match(string, pos=pos)
+ if m:
+ skip_lines = m.group(0).split('\n')
+ if len(skip_lines) > 1:
+ self.row += len(skip_lines) - 1
+ self.column = 1 + len(skip_lines[-1])
+ else:
+ self.column += len(skip_lines[0])
+
+ pos = m.end()
+ continue
+
+ for cls, type, regex in self.token_map:
+ m = regex.match(string, pos=pos)
+ if m:
+ yield cls(type, m.group(0),
+ self.filename, self.row, self.column)
+ self.column += len(m.group(0))
+ pos = m.end()
+ break
+ else:
+ raise ConfigParseError(
+ "Couldn't load config in %r row %d, column %d: %r" %
+ (self.filename, self.row, self.column,
+ string[pos:pos+20]))
+
+
+class TokenStream:
+ '''Offer a parsing-oriented view on tokens
+
+ Provide several methods that are useful to parsers, like ``accept()``,
+ ``expect()``, ...
+
+ The ``from_file()`` method is the preferred way to read input files, as
+ it handles include directives, which the ``Tokenizer`` class does not do.
+ '''
+
+ def __init__(self, tokens):
+ self.position = 0
+ self.tokens = list(tokens)
+
+ @classmethod
+ def from_file(cls, f, filename=None, includedir='', seenfiles=None):
+ '''Create a token stream by reading an input file
+
+ Read tokens from `f`. If an include directive ('@include "file.cfg"')
+ is found, read its contents as well.
+
+ The `filename` argument is used for error messages and to detect
+ circular imports. ``includedir`` sets the lookup directory for included
+ files. ``seenfiles`` is used internally to detect circular includes,
+ and should normally not be supplied by users of is function.
+ '''
+
+ if filename is None:
+ filename = getattr(f, 'name', '<unknown>')
+ if seenfiles is None:
+ seenfiles = set()
+
+ if filename in seenfiles:
+ raise ConfigParseError("Circular include: %r" % (filename,))
+ seenfiles = seenfiles | {filename} # Copy seenfiles, don't alter it.
+
+ tokenizer = Tokenizer(filename=filename)
+ lines = []
+ tokens = []
+ for line in f:
+ m = re.match(r'@include "(.*)"$', line.strip())
+ if m:
+ tokens.extend(tokenizer.tokenize(''.join(lines)))
+ lines = [re.sub(r'\S', ' ', line)]
+
+ includefilename = decode_escapes(m.group(1))
+ includefilename = os.path.join(includedir, includefilename)
+ try:
+ includefile = open(includefilename, "r")
+ except IOError:
+ raise ConfigParseError("Could not open include file %r" %
+ (includefilename,))
+
+ with includefile:
+ includestream = cls.from_file(includefile,
+ filename=includefilename,
+ includedir=includedir,
+ seenfiles=seenfiles)
+ tokens.extend(includestream.tokens)
+
+ else:
+ lines.append(line)
+
+ tokens.extend(tokenizer.tokenize(''.join(lines)))
+ return cls(tokens)
+
+ def peek(self):
+ '''Return (but do not consume) the next token
+
+ At the end of input, ``None`` is returned.
+ '''
+
+ if self.position >= len(self.tokens):
+ return None
+
+ return self.tokens[self.position]
+
+ def accept(self, *args):
+ '''Consume and return the next token if it has the correct type
+
+ Multiple token types (as strings, e.g. 'integer64') can be given
+ as arguments. If the next token is one of them, consume and return it.
+
+ If the token type doesn't match, return None.
+ '''
+
+ token = self.peek()
+ if token is None:
+ return None
+
+ for arg in args:
+ if token.type == arg:
+ self.position += 1
+ return token
+
+ return None
+
+ def expect(self, *args):
+ '''Consume and return the next token if it has the correct type
+
+ Multiple token types (as strings, e.g. 'integer64') can be given
+ as arguments. If the next token is one of them, consume and return it.
+
+ If the token type doesn't match, raise a ConfigParseError.
+ '''
+
+ t = self.accept(*args)
+ if t is not None:
+ return t
+
+ self.error("expected: %r" % (args,))
+
+ def error(self, msg):
+ '''Raise a ConfigParseError at the current input position'''
+ if self.finished():
+ raise ConfigParseError("Unexpected end of input; %s" % (msg,))
+ else:
+ t = self.peek()
+ raise ConfigParseError("Unexpected token %s; %s" % (t, msg))
+
+ def finished(self):
+ '''Return ``True`` if the end of the token stream is reached.'''
+ return self.position >= len(self.tokens)
+
+
+class Parser:
+ '''Recursive descent parser for libconfig files
+
+ Takes a ``TokenStream`` as input, the ``parse()`` method then returns
+ the config file data in a ``json``-module-style format.
+ '''
+
+ def __init__(self, tokenstream):
+ self.tokens = tokenstream
+
+ def parse(self):
+ return self.configuration()
+
+ def configuration(self):
+ result = self.setting_list_or_empty()
+ if not self.tokens.finished():
+ raise ConfigParseError("Expected end of input but found %s" %
+ (self.tokens.peek(),))
+
+ return result
+
+ def setting_list_or_empty(self):
+ result = AttrDict()
+ while True:
+ s = self.setting()
+ if s is None:
+ return result
+
+ result[s[0]] = s[1]
+
+ def setting(self):
+ name = self.tokens.accept('name')
+ if name is None:
+ return None
+
+ self.tokens.expect(':', '=')
+
+ value = self.value()
+ if value is None:
+ self.tokens.error("expected a value")
+
+ self.tokens.accept(';', ',')
+
+ return (name.text, value)
+
+ def value(self):
+ acceptable = [self.scalar_value, self.array, self.list, self.group]
+ return self._parse_any_of(acceptable)
+
+ def scalar_value(self):
+ # This list is ordered so that more common tokens are checked first.
+ acceptable = [self.string, self.boolean, self.integer, self.float,
+ self.hex, self.integer64, self.hex64]
+ return self._parse_any_of(acceptable)
+
+ def value_list_or_empty(self):
+ return tuple(self._comma_separated_list_or_empty(self.value))
+
+ def scalar_value_list_or_empty(self):
+ return self._comma_separated_list_or_empty(self.scalar_value)
+
+ def array(self):
+ return self._enclosed_block('[', self.scalar_value_list_or_empty, ']')
+
+ def list(self):
+ return self._enclosed_block('(', self.value_list_or_empty, ')')
+
+ def group(self):
+ return self._enclosed_block('{', self.setting_list_or_empty, '}')
+
+ def boolean(self):
+ return self._create_value_node('boolean')
+
+ def integer(self):
+ return self._create_value_node('integer')
+
+ def integer64(self):
+ return self._create_value_node('integer64')
+
+ def hex(self):
+ return self._create_value_node('hex')
+
+ def hex64(self):
+ return self._create_value_node('hex64')
+
+ def float(self):
+ return self._create_value_node('float')
+
+ def string(self):
+ t_first = self.tokens.accept('string')
+ if t_first is None:
+ return None
+
+ values = [t_first.value]
+ while True:
+ t = self.tokens.accept('string')
+ if t is None:
+ break
+ values.append(t.value)
+
+ return ''.join(values)
+
+ def _create_value_node(self, tokentype):
+ t = self.tokens.accept(tokentype)
+ if t is None:
+ return None
+
+ return t.value
+
+ def _parse_any_of(self, nonterminals):
+ for fun in nonterminals:
+ result = fun()
+ if result is not None:
+ return result
+
+ return None
+
+ def _comma_separated_list_or_empty(self, nonterminal):
+ values = []
+ first = True
+ while True:
+ v = nonterminal()
+ if v is None:
+ if first:
+ return []
+ else:
+ # This is disabled to enable the last member in a list to have a comma at the end
+ # self.tokens.error("expected value after ','")
+ return values
+
+ values.append(v)
+ if not self.tokens.accept(','):
+ return values
+
+ first = False
+
+ def _enclosed_block(self, start, nonterminal, end):
+ if not self.tokens.accept(start):
+ return None
+ result = nonterminal()
+ self.tokens.expect(end)
+ return result
+
+
+def load(f, filename=None, includedir=''):
+ '''Load the contents of ``f`` (a file-like object) to a Python object
+
+ The returned object is a subclass of ``dict`` that exposes string keys as
+ attributes as well.
+
+ Example:
+
+ >>> with open('test/example.cfg') as f:
+ ... config = libconf.load(f)
+ >>> config['window']['title']
+ 'libconfig example'
+ >>> config.window.title
+ 'libconfig example'
+ '''
+
+ if isinstance(f.read(0), bytes):
+ raise TypeError("libconf.load() input file must by unicode")
+
+ tokenstream = TokenStream.from_file(f,
+ filename=filename,
+ includedir=includedir)
+ return Parser(tokenstream).parse()
+
+
+def loads(string, filename=None, includedir=''):
+ '''Load the contents of ``string`` to a Python object
+
+ The returned object is a subclass of ``dict`` that exposes string keys as
+ attributes as well.
+
+ Example:
+
+ >>> config = libconf.loads('window: { title: "libconfig example"; };')
+ >>> config['window']['title']
+ 'libconfig example'
+ >>> config.window.title
+ 'libconfig example'
+ '''
+
+ try:
+ f = io.StringIO(string)
+ except TypeError:
+ raise TypeError("libconf.loads() input string must by unicode")
+
+ return load(f, filename=filename, includedir=includedir)
+
+
+# dump() logic
+##############
+
+def dump_int(i):
+ '''Stringize ``i``, append 'L' if ``i`` is exceeds the 32-bit int range'''
+ return str(i) + ('' if SMALL_INT_MIN <= i <= SMALL_INT_MAX else 'L')
+
+
+def dump_string(s):
+ '''Stringize ``s``, adding double quotes and escaping as necessary
+
+ Backslash escape backslashes, double quotes, ``\f``, ``\n``, ``\r``, and
+ ``\t``. Escape all remaining unprintable characters in ``\xFF``-style.
+ The returned string will be surrounded by double quotes.
+ '''
+
+ s = (s.replace('\\', '\\\\')
+ .replace('"', '\\"')
+ .replace('\f', r'\f')
+ .replace('\n', r'\n')
+ .replace('\r', r'\r')
+ .replace('\t', r'\t'))
+ s = UNPRINTABLE_CHARACTER_RE.sub(
+ lambda m: r'\x{:02x}'.format(ord(m.group(0))),
+ s)
+ return '"' + s + '"'
+
+
+def dump_value(key, value, f, indent=0):
+ '''Save a value of any libconfig type
+
+ This function serializes takes ``key`` and ``value`` and serializes them
+ into ``f``. If ``key`` is ``None``, a list-style output is produced.
+ Otherwise, output has ``key = value`` format.
+ '''
+
+ spaces = ' ' * indent
+
+ if key is None:
+ key_prefix = ''
+ key_prefix_nl = ''
+ else:
+ key_prefix = key + ' = '
+ key_prefix_nl = key + ' =\n' + spaces
+
+ if isinstance(value, dict):
+ f.write(u'{}{}{{\n'.format(spaces, key_prefix_nl))
+ dump_dict(value, f, indent + 4)
+ f.write(u'{}}}'.format(spaces))
+ elif isinstance(value, tuple):
+ f.write(u'{}{}(\n'.format(spaces, key_prefix_nl))
+ dump_collection(value, f, indent + 4)
+ f.write(u'\n{})'.format(spaces))
+ elif isinstance(value, list):
+ f.write(u'{}{}[\n'.format(spaces, key_prefix_nl))
+ dump_collection(value, f, indent + 4)
+ f.write(u'\n{}]'.format(spaces))
+ elif isstr(value):
+ f.write(u'{}{}{}'.format(spaces, key_prefix, dump_string(value)))
+ elif isint(value):
+ f.write(u'{}{}{}'.format(spaces, key_prefix, dump_int(value)))
+ elif isinstance(value, float):
+ f.write(u'{}{}{}'.format(spaces, key_prefix, value))
+ else:
+ raise ConfigSerializeError("Can not serialize object %r of type %s" %
+ (value, type(value)))
+
+
+def dump_collection(cfg, f, indent=0):
+ '''Save a collection of attributes'''
+
+ for i, value in enumerate(cfg):
+ dump_value(None, value, f, indent)
+ if i < len(cfg) - 1:
+ f.write(u',\n')
+
+
+def dump_dict(cfg, f, indent=0):
+ '''Save a dictionary of attributes'''
+
+ for key in cfg:
+ if not isstr(key):
+ raise ConfigSerializeError("Dict keys must be strings: %r" %
+ (key,))
+ dump_value(key, cfg[key], f, indent)
+ f.write(u';\n')
+
+
+def dumps(cfg):
+ '''Serialize ``cfg`` into a libconfig-formatted ``str``
+
+ ``cfg`` must be a ``dict`` with ``str`` keys and libconf-supported values
+ (numbers, strings, booleans, possibly nested dicts, lists, and tuples).
+
+ Returns the formatted string.
+ '''
+
+ str_file = io.StringIO()
+ dump(cfg, str_file)
+ return str_file.getvalue()
+
+
+def dump(cfg, f):
+ '''Serialize ``cfg`` as a libconfig-formatted stream into ``f``
+
+ ``cfg`` must be a ``dict`` with ``str`` keys and libconf-supported values
+ (numbers, strings, booleans, possibly nested dicts, lists, and tuples).
+
+ ``f`` must be a ``file``-like object with a ``write()`` method.
+ '''
+
+ if not isinstance(cfg, dict):
+ raise ConfigSerializeError(
+ 'dump() requires a dict as input, not %r of type %r' %
+ (cfg, type(cfg)))
+
+ dump_dict(cfg, f, 0)
+
+
+# main(): small example of how to use libconf
+#############################################
+
+def main():
+ '''Open the libconfig file specified by sys.argv[1] and pretty-print it'''
+ global output
+ if len(sys.argv[1:]) == 1:
+ with io.open(sys.argv[1], 'r', encoding='utf-8') as f:
+ output = load(f)
+ else:
+ output = load(sys.stdin)
+
+ dump(output, sys.stdout)
+
+
+if __name__ == '__main__':
+ main()