remove_estimate.py

#   Copyright 2017 OmiseGO Pte Ltd
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.

import json

import click

from utils import remove_estimate


@click.command()
@click.argument('unsigned-file', type=click.File('rb'))
@click.argument('unsigned-file-out', type=click.File('wb'))
def remove(unsigned_file, unsigned_file_out):
    unsigned = json.loads(unsigned_file.read())

    unsigned_out = remove_estimate(unsigned)

    unsigned_file_out.write(json.dumps(unsigned_out, sort_keys=True))


if __name__ == '__main__':
    remove()