diff options
| author | nalpp <[email protected]> | 2022-02-10 16:46:45 +0300 | 
|---|---|---|
| committer | Daniil Cherednik <[email protected]> | 2022-02-10 16:46:45 +0300 | 
| commit | 87f3eb38999df2d3c1cb77f8ffb9c52ec9c516fb (patch) | |
| tree | f4ab0734eecf35b22f21952d1aa15496f267a6a3 /build/scripts/python_yndexer.py | |
| parent | 198698b9c1e18958addc9682fbfba4576d604757 (diff) | |
Restoring authorship annotation for <[email protected]>. Commit 1 of 2.
Diffstat (limited to 'build/scripts/python_yndexer.py')
| -rw-r--r-- | build/scripts/python_yndexer.py | 100 | 
1 files changed, 50 insertions, 50 deletions
diff --git a/build/scripts/python_yndexer.py b/build/scripts/python_yndexer.py index 3180665387d..2ffcb104f4d 100644 --- a/build/scripts/python_yndexer.py +++ b/build/scripts/python_yndexer.py @@ -1,53 +1,53 @@ -import os -import sys -import threading -import subprocess - - -def _try_to_kill(process): -    try: -        process.kill() -    except Exception: -        pass - - -def touch(path): -    if not os.path.exists(path): -        with open(path, 'w') as _: -            pass - - -class Process(object): -    def __init__(self, args): -        self._process = subprocess.Popen(args) -        self._event = threading.Event() -        self._result = None -        thread = threading.Thread(target=self._run) -        thread.setDaemon(True) -        thread.start() - -    def _run(self): -        self._process.communicate() -        self._result = self._process.returncode -        self._event.set() - -    def wait(self, timeout): -        self._event.wait(timeout=timeout) -        _try_to_kill(self._process) -        return self._result - - -if __name__ == '__main__': -    yndexer = sys.argv[1] -    timeout = int(sys.argv[2]) -    output_file = sys.argv[3] -    input_file = sys.argv[4] +import os  +import sys  +import threading  +import subprocess  +  +  +def _try_to_kill(process):  +    try:  +        process.kill()  +    except Exception:  +        pass  +  +  +def touch(path):  +    if not os.path.exists(path):  +        with open(path, 'w') as _:  +            pass  +  +  +class Process(object):  +    def __init__(self, args):  +        self._process = subprocess.Popen(args)  +        self._event = threading.Event()  +        self._result = None  +        thread = threading.Thread(target=self._run)  +        thread.setDaemon(True)  +        thread.start()  +  +    def _run(self):  +        self._process.communicate()  +        self._result = self._process.returncode  +        self._event.set()  +  +    def wait(self, timeout):  +        self._event.wait(timeout=timeout)  +        _try_to_kill(self._process)  +        return self._result  +  +  +if __name__ == '__main__':  +    yndexer = sys.argv[1]  +    timeout = int(sys.argv[2])  +    output_file = sys.argv[3]  +    input_file = sys.argv[4]       partition_count = sys.argv[5]      partition_index = sys.argv[6] - +       process = Process([yndexer, '-f', input_file, '-y', output_file, '-c', partition_count, '-i', partition_index]) -    result = process.wait(timeout=timeout) - -    if result != 0: -        print >> sys.stderr, 'Yndexing process finished with code', result -        touch(output_file) +    result = process.wait(timeout=timeout)  +  +    if result != 0:  +        print >> sys.stderr, 'Yndexing process finished with code', result  +        touch(output_file)   | 
