dda-python-terraform/python_terraform/tfstate.py
Freddy Tan 2ff59f3630 1. add test cases, try tdd
2. refactor to more generic method instead of aws method
3. ready for release to pypi
2016-11-18 15:35:14 +08:00

31 lines
No EOL
802 B
Python

import json
import os
import logging
log = logging.getLogger(__name__)
class Tfstate(object):
def __init__(self, data=None):
self.tfstate_file = None
self.native_data = data
if data:
self.__dict__ = data
@staticmethod
def load_file(file_path):
"""
Read the tfstate file and load its contents, parses then as JSON and put the result into the object
"""
log.debug('read data from {0}'.format(file_path))
if os.path.exists(file_path):
with open(file_path) as f:
json_data = json.load(f)
tf_state = Tfstate(json_data)
tf_state.tfstate_file = file_path
return tf_state
log.warn('{0} is not exist'.format(file_path))
return Tfstate()