scrap-is-not-scrap/run.py

#! /usr/bin/env python3
# encoding: utf-8
import os
import logging
import subprocess
import json


class Sins():
    logger = logging.getLogger()

    def __init__(self, parent, seed=None, run_dir=None):
        self.logger.info('execute\n{}'.format(json.dumps(
            (parent, seed, run_dir))
        ))

        if os.path.isdir(run_dir):
            self.run_dir = os.path.abspath(run_dir)

        self.seed = os.path.abspath(seed)

        parent = os.path.abspath(parent)
        if not os.path.isfile(parent):
            raise ValueError('Invalid executable image path.')

        self.parent = parent

        parsed = self.parent, self.seed, self.run_dir
        self.logger.info('parsed\n{}'.format(json.dumps(parsed)))

        while True:
            self.execute(self.parent, self.seed, self.run_dir)
            paths = sorted(os.listdir(self.run_dir))
            for path in paths:
                scrap_path = os.path.join(self.run_dir, path)
                if os.path.isfile(scrap_path):
                    self.execute(
                        self.parent,
                        scrap_path,
                        self.run_dir
                    )

    def scrap_recent(self, run_dir):
        scraps = sorted(os.listdir(run_dir))

        if scraps:
            return os.path.join(run_dir, scraps[-1])

        return None

    def execute(self, parent, scrap, cwd):
        self.logger.info('execute\n{}'.format(
            json.dumps((parent, scrap, cwd))
        ))
        subprocess.run([parent, scrap], cwd=cwd)


def hex_dumps(scrap_dir):
    scrap_dir = os.path.abspath(scrap_dir)
    dump_dir = os.path.join(scrap_dir, 'hexdumps')

    try:
        os.mkdir(dump_dir)
    except:
        pass

    scraps = os.listdir(scrap_dir)

    for scrap in scraps:
        scrap_path = os.path.join(scrap_dir, scrap)
        hex_name = '{}.hex'.format(scrap)
        hex_path = os.path.join(dump_dir, hex_name)

        if os.path.isfile(scrap_path):
            with open(scrap_path, 'rb') as file_in:
                with open(hex_path, 'w') as file_out:
                    binary = bytes(file_in.read())
                    for each in binary:
                        file_out.writelines('\'\\x{:02X}\',\n'.format(each))


if __name__ == '__main__':
    import argparse

    logger = logging.getLogger()
    logger.setLevel(logging.DEBUG)

    file_handler = logging.FileHandler(filename='sins.log')
    formatter = logging.Formatter(
        '#%(asctime)s %(name)s %(levelname)s\n##%(message)s\n'
    )
    file_handler.setFormatter(formatter)
    logger.addHandler(file_handler)

    parser = argparse.ArgumentParser()
    parser.add_argument('-seed', default='build/scrap.asm.2.o')
    parser.add_argument('-parent', default='build/sins')
    parser.add_argument('-dir', default='sandbox')
    parser.add_argument('-dumps', action='store_true')
    parser.add_argument('-build', action='store_true')
    args = parser.parse_args()

    if args.build:
        waf = ['python3', 'waf.py', 'configure', 'build']
        waf_proc = subprocess.run(waf, stdout=subprocess.PIPE)
        logger.info(waf_proc.stdout)

    if args.dumps:
        hex_dumps(args.dir)
    else:
        sins = Sins(parent=args.parent, seed=args.seed, run_dir=args.dir)