nemunaire/nemubot
1
0
Fork 0
Code Releases Activity

XML datastore: load will now automatically try to load backup

Browse source
This commit is contained in:
nemunaire 2015-06-13 14:00:04 +02:00
parent ab2eb405ca
commit c7706bfc97
1 changed files with 23 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

22
nemubot/datastore/xml.py
View file

@ -15,11 +15,14 @@
# along with this program. If not, see <http://www.gnu.org/licenses/>. # along with this program. If not, see <http://www.gnu.org/licenses/>.
import fcntl import fcntl
import logging
import os import os
import xml.parsers.expat import xml.parsers.expat
from nemubot.datastore.abstract import Abstract from nemubot.datastore.abstract import Abstract
logger = logging.getLogger("nemubot.datastore.xml")
class XML(Abstract): class XML(Abstract):
@ -88,10 +91,27 @@ class XML(Abstract):
""" """
data_file = self._get_data_file_path(module) data_file = self._get_data_file_path(module)
# Try to load original file
if os.path.isfile(data_file): if os.path.isfile(data_file):
from nemubot.tools.xmlparser import parse_file from nemubot.tools.xmlparser import parse_file
try:
return parse_file(data_file) return parse_file(data_file)
else: except xml.parsers.expat.ExpatError:
# Try to load from backup
for i in range(10):
path = data_file + "." + str(i)
if os.path.isfile(path):
try:
cnt = parse_file(path)
logger.warn("Restoring from backup: %s", path)
return cnt
except xml.parsers.expat.ExpatError:
continue
# Default case: initialize a new empty datastore
return Abstract.load(self, module) return Abstract.load(self, module)
def _rotate(self, path): def _rotate(self, path):