dynamic_building_simulation/insel/insel.py

import os
from pathlib import Path


class Insel:
  def __init__(self, path, name, new_content="", mode=1, keep_files=False):
    self._path = path
    self._name = name
    self._full_path = None
    self._content = None
    self._results = None
    self._keep_files = keep_files
    self.add_content(new_content, mode)
    self.save()
    self.run()

  def save(self):
    with open(self.full_path, 'w') as insel_file:
      insel_file.write(self.content)
    return

  @property
  def full_path(self):
    if self._full_path is None:
      self._full_path = (Path(self._path) / 'third_party_files/insel' / self._name).resolve()
    return self._full_path

  @property
  def content(self):
    if self._content is None:
      if os.path.exists(self.full_path):
        with open(self.full_path, 'r') as insel_file:
          self._content = insel_file.read()
      else:
        self._content = ''
    return self._content

  @staticmethod
  def add_block(file, block_number, block_type, inputs='', parameters=''):
    file += "S " + str(block_number) + " " + block_type + "\r\n"
    for block_input in inputs:
      file += block_input + "\r\n"
    if len(parameters) > 0:
      file += "P " + str(block_number) + "\r\n"
    for block_parameter in parameters:
      file += block_parameter + "\r\n"
    return file

  def add_content(self, new_content, mode):
    # mode = 1: keep old content
    if mode == 1:
      self._content = self.content + '\r\n' + new_content
    # mode = 2: over-write
    elif mode == 2:
      self._content = new_content
    else:
      raise Exception('Add content mode not supported')
    return

  def run(self):
    finish = os.system('insel ' + str(self.full_path))
    os.close(finish)
    if not self._keep_files:
      os.remove(self.full_path)

  @property
  def results(self):
    raise Exception('Not implemented')