First commit

Author: beskep

.gitignore

@@ -0,0 +1,143 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# project
+tests/data/
+.vscode/
+*.cs

ECO2/cli.py

@@ -0,0 +1,62 @@
+import click
+
+from eco2 import Eco2
+
+
+@click.group()
+def cli():
+    pass
+
+
+@cli.command()
+@click.argument('eco')
+@click.option('-s',
+              '--savedir',
+              help=('해석한 header와 value 파일의 저장 경로. '
+                    '미설정 시 eco 파일과 동일 경로에 저장.'))
+@click.option('-h',
+              '--header',
+              default='header',
+              help='header 파일 이름. 미설정 시 `header`로 지정.')
+@click.option('-v',
+              '--value',
+              help=('value 파일 이름. '
+                    '미설정 시 eco 파일과 동일 이름으로 저장.'))
+def decrypt(eco, savedir, header, value):
+    """
+    \b
+    eco 파일을 해석해서 header와 value 파일로 나눠 저장.
+        header: eco의 버전, 생성 날짜 등 정보를 담은 바이너리 파일.
+        value: 해석 설정 정보를 담은 xml 파일.
+
+    \b
+    Argument:
+        ECO: 해석할 eco 파일 경로
+    """
+    Eco2.decrypt(path=eco,
+                 save_dir=savedir,
+                 header_name=header,
+                 value_name=value)
+
+
+@cli.command()
+@click.argument('header')
+@click.argument('value')
+@click.option('-s',
+              '--savedir',
+              help=('해석한 header와 value 파일의 저장 경로. '
+                    '미설정 시 eco 파일과 동일 경로에 저장.'))
+def encrypt(header, value, savedir):
+    """
+    header와 value를 암호화해서 eco 파일로 변환
+
+    \b
+    Arguments:
+        HEADER: header 파일 경로.
+        VALUE : value 파일 경로. 폴더 경로를 지정하면 해당 폴더에 존재하는 모든 .xml 파일을 변환.
+    """
+    Eco2.encrypt_dir(header_path=header, value_path=value, save_dir=savedir)
+
+
+if __name__ == '__main__':
+    cli()

ECO2/cx_setup.py

@@ -0,0 +1,23 @@
+from cx_Freeze import Executable
+from cx_Freeze import setup
+
+if __name__ == '__main__':
+    options = {
+        'build_exe': {
+            'optimize':
+                1,
+            'excludes': [
+                'email', 'html', 'http', 'logging', 'tkinter', 'unittest'
+            ]
+        }
+    }
+
+    executables = [
+        Executable(script=r'ECO2\cli.py', target_name='ECO2'),
+    ]
+
+    setup(name='ECO2',
+          version='0.1',
+          description='ECO2',
+          options=options,
+          executables=executables)

ECO2/eco2.py

@@ -0,0 +1,130 @@
+from itertools import cycle
+from pathlib import Path
+from typing import Union
+
+
+class Eco2:
+    header = (
+        (2, 'SF_type'),
+        (10, 'UI_version'),
+        (10, 'LG_version'),
+        (100, 'name'),
+        (256, 'desc'),
+        (19, 'make_time'),
+        (19, 'edit_time'),
+        (8, 'unknown'),
+    )
+    key = (172, 41, 85, 66)
+    encoding = 'UTF-8'
+    value_ext = '.xml'
+
+    @classmethod
+    def decrypt_bytes(cls, data: bytes):
+        return bytes((d ^ k for d, k in zip(data, cycle(cls.key))))
+
+    @classmethod
+    def encrypt_bytes(cls, data: bytes):
+        return cls.decrypt_bytes(data)
+
+    @classmethod
+    def header_length(cls):
+        return sum([x[0] for x in cls.header])
+
+    @staticmethod
+    def _decode_chunk(b: bytes, length: int):
+        data = b[:length]
+        bnext = b[length:]
+
+        return data, bnext
+
+    @classmethod
+    def _decode_header(cls, data: bytes):
+        header = {}
+        b = data
+        for length, name in cls.header:
+            value, b = cls._decode_chunk(b=b, length=length)
+            header[name] = value
+
+        return header
+
+    @classmethod
+    def _decrypt_eco2_data(cls, data: bytes):
+        decrypted = cls.decrypt_bytes(data)
+        hl = cls.header_length()
+
+        header_bytes = decrypted[:hl]
+        value_bytes = decrypted[hl:]
+        value = value_bytes.decode()
+
+        return header_bytes, value
+
+    @classmethod
+    def _write_value(cls, path: Path, value: str):
+        path.write_text(value.replace('\r\n', '\n'), encoding=cls.encoding)
+
+    @classmethod
+    def _read_value(cls, path: Path):
+        return path.read_text(encoding=cls.encoding).replace('\n', '\r\n')
+
+    @classmethod
+    def decrypt(cls,
+                path: Union[str, Path],
+                save_dir=None,
+                header_name=None,
+                value_name=None):
+        path = Path(path)
+        save_dir = path.parent if save_dir is None else Path(save_dir)
+        if not save_dir.exists():
+            raise FileNotFoundError(save_dir)
+
+        if header_name is None:
+            header_name = 'header'
+        if value_name is None:
+            value_name = path.stem
+
+        data = path.read_bytes()
+        bheader, value = cls._decrypt_eco2_data(data)
+
+        header_path = save_dir.joinpath(header_name)
+        header_path.write_bytes(bheader)
+
+        value_path = save_dir.joinpath(value_name + cls.value_ext)
+        cls._write_value(path=value_path, value=value)
+
+    @classmethod
+    def encrypt(cls, header_path, value_path, save_path=None):
+        if save_path is None:
+            save_path = 'output.eco'
+
+        header_path = Path(header_path)
+        value_path = Path(value_path)
+        save_path = Path(save_path)
+
+        bheader = header_path.read_bytes()
+
+        value = cls._read_value(path=value_path)
+        bvalue = value.encode()
+
+        data = bheader + bvalue
+        encrypted = cls.encrypt_bytes(data)
+
+        save_path.write_bytes(encrypted)
+
+    @classmethod
+    def encrypt_dir(cls, header_path, value_path, save_dir=None):
+        header_path = Path(header_path)
+        value_path = Path(value_path)
+
+        save_dir = value_path if save_dir is None else Path(save_dir)
+        if not save_dir.is_dir():
+            save_dir = save_dir.parent
+
+        if value_path.is_dir():
+            vps = value_path.glob(f'*{cls.value_ext}')
+        else:
+            vps = [value_path]
+
+        for vp in vps:
+            cls.encrypt(header_path=header_path,
+                        value_path=vp,
+                        save_path=save_dir.joinpath(f'{vp.stem}.eco'))

README.md

@@ -0,0 +1,3 @@
+# ECO2
+
+효율등급인증 평가프로그램 ECO2의 저장 파일 (`.eco`) 해석