src/nepi/util/serializer.py

   1 #
   2 #    NEPI, a framework to manage network experiments
   3 #    Copyright (C) 2013 INRIA
   4 #
   5 #    This program is free software: you can redistribute it and/or modify
   6 #    it under the terms of the GNU General Public License as published by
   7 #    the Free Software Foundation, either version 3 of the License, or
   8 #    (at your option) any later version.
   9 #
  10 #    This program is distributed in the hope that it will be useful,
  11 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 #    GNU General Public License for more details.
  14 #
  15 #    You should have received a copy of the GNU General Public License
  16 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 #
  18 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
  19
  20 import datetime
  21 import os
  22
  23 class SFormats:
  24     XML = "xml"
  25
  26 class ECSerializer(object):
  27     def load(self, filepath, format = SFormats.XML):
  28         if format == SFormats.XML:
  29             from nepi.util.parsers.xml_parser import ECXMLParser
  30
  31             parser = ECXMLParser()
  32             f = open(filepath, "r")
  33             xml = f.read()
  34             f.close()
  35
  36             ec = parser.from_xml(xml)
  37
  38         return ec
  39
  40     def serialize(self, ec, format = SFormats.XML):
  41         if format == SFormats.XML:
  42             from nepi.util.parsers.xml_parser import ECXMLParser
  43
  44             parser = ECXMLParser()
  45             sec = parser.to_xml(ec)
  46
  47         return sec
  48
  49     def save(self, ec, dirpath, format = SFormats.XML):
  50         date = datetime.datetime.now().strftime('%Y%m%d%H%M%S')
  51         filename = "%s_%s" % (ec.exp_id, date)
  52
  53         if format == SFormats.XML:
  54             filepath = os.path.join(dirpath, "%s.xml" % filename)
  55             sec = self.serialize(ec, format = format)
  56             f = open(filepath, "w")
  57             f.write(sec)
  58             f.close()
  59
  60         return filepath
  61