Merge pull request #1 from pipermerriam/piper/initial-implementation

Initial implementation

Author: pipermerriam

.travis.yml

@@ -6,13 +6,9 @@ sudo: required
 before_install:
   - travis_retry sudo add-apt-repository -y ppa:ethereum/ethereum
   - travis_retry sudo apt-get update
-  - travis_retry sudo apt-get install -y ethereum
-  - mkdir -p ~/.ethash
-  - geth makedag 0 ~/.ethash
+  - travis_retry sudo apt-get install -y solc
 cache:
   pip: true
-  directories:
-    - ~/.ethash
 env:
   matrix:
     - TOX_ENV=py27

README.md

@@ -37,7 +37,7 @@ pip install py-solc
         },
     },
 }
->>> compile_source(["/path/to/Foo.sol", "/path/to/Bar.sol"])
+>>> compile_files(["/path/to/Foo.sol", "/path/to/Bar.sol"])
 {
     'Foo': {
         'abi': [{'inputs': [], 'type': 'constructor'}],

conftest.py

@@ -0,0 +1,6 @@
+import pytest
+
+
+@pytest.fixture()
+def contracts_dir(tmpdir):
+    return str(tmpdir.mkdir("contracts"))

requirements-dev.txt

@@ -0,0 +1,2 @@
+pytest>=2.9.2
+tox>=2.3.1

requirements.txt

@@ -0,0 +1,7 @@
+from __future__ import absolute_import
+
+from .main import (  # NOQA
+    get_solc_version,
+    compile_files,
+    compile_source,
+)

solc/__init__.py

@@ -0,0 +1,6 @@
+class SolcError(Exception):
+    pass
+
+
+class CompileError(Exception):
+    pass

solc/exceptions.py

@@ -0,0 +1,123 @@
+from __future__ import absolute_import
+
+import functools
+import json
+import re
+
+from .exceptions import (
+    SolcError,
+)
+
+from .utils.formatting import (
+    add_0x_prefix,
+)
+from .utils.filesystem import (
+    is_executable_available,
+)
+from .wrapper import (
+    SOLC_BINARY,
+    solc_wrapper,
+)
+
+
+version_regex = re.compile('Version: ([0-9]+\.[0-9]+\.[0-9]+(-[a-f0-9]+)?)')
+
+
+is_solc_available = functools.partial(is_executable_available, SOLC_BINARY)
+
+
+def get_solc_version():
+    stdoutdata, stderrdata = solc_wrapper(version=True)
+    version_match = version_regex.search(stdoutdata)
+    if version_match is None:
+        raise SolcError(
+            "Unable to extract version string from command output: `{0}`".format(
+                stdoutdata,
+            )
+        )
+    return version_match.groups()[0]
+
+
+def _parse_compiler_output(stdoutdata):
+    contracts = json.loads(stdoutdata)['contracts']
+
+    for _, data in contracts.items():
+        data['abi'] = json.loads(data['abi'])
+
+    sorted_contracts = sorted(contracts.items(), key=lambda c: c[0])
+
+    compiler_version = get_solc_version()
+
+    return {
+        contract_name: {
+            'abi': contract_data['abi'],
+            'code': add_0x_prefix(contract_data['bin']),
+            'code_runtime': add_0x_prefix(contract_data['bin-runtime']),
+            'source': None,
+            'meta': {
+                'compilerVersion': compiler_version,
+                'language': 'Solidity',
+                'languageVersion': '0',
+            },
+        }
+        for contract_name, contract_data
+        in sorted_contracts
+    }
+
+
+ALL_OUTPUT_VALUES = [
+    "abi",
+    "asm",
+    "ast",
+    "bin",
+    "bin-runtime",
+    "clone-bin",
+    "devdoc",
+    "interface",
+    "opcodes",
+    "userdoc",
+]
+
+
+def compile_source(source, output_values=ALL_OUTPUT_VALUES, **kwargs):
+    if 'stdin_bytes' in kwargs:
+        raise ValueError(
+            "The `stdin_bytes` keyword is not allowed in the `compile_source` function"
+        )
+    if 'combined_json' in kwargs:
+        raise ValueError(
+            "The `combined_json` keyword is not allowed in the `compile_source` function"
+        )
+
+    combined_json = ','.join(output_values)
+
+    stdoutdata, stderrdata = solc_wrapper(
+        stdin_bytes=source,
+        combined_json=combined_json,
+        **kwargs
+    )
+
+    contracts = _parse_compiler_output(stdoutdata)
+    return contracts
+
+
+def compile_files(source_files, output_values=ALL_OUTPUT_VALUES, **kwargs):
+    if 'source_files' in kwargs:
+        raise ValueError(
+            "The `source_files` keyword is not allowed in the `compile_files` function"
+        )
+    if 'combined_json' in kwargs:
+        raise ValueError(
+            "The `combined_json` keyword is not allowed in the `compile_files` function"
+        )
+
+    combined_json = ','.join(output_values)
+
+    stdoutdata, stderrdata = solc_wrapper(
+        source_files=source_files,
+        combined_json=combined_json,
+        **kwargs
+    )
+
+    contracts = _parse_compiler_output(stdoutdata)
+    return contracts

solc/main.py

@@ -0,0 +1,19 @@
+import os
+
+
+def is_executable_available(program):
+    def is_exe(fpath):
+        return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
+
+    fpath = os.path.dirname(program)
+    if fpath:
+        if is_exe(program):
+            return True
+    else:
+        for path in os.environ["PATH"].split(os.pathsep):
+            path = path.strip('"')
+            exe_file = os.path.join(path, program)
+            if is_exe(exe_file):
+                return True
+
+    return False