| OLD | NEW |
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 # -*- coding: utf-8 -*- ------------------------------------------------ | |
| 3 # vi:set fenc=utf-8 ft=python ts=8 et sw=4 sts=4: | |
| 4 | |
| 5 __doc__ = """ Redirect STDIN into an ABP log channel. | |
| 6 | |
| 7 """ | |
| 8 import argparse | |
| 9 import fcntl | |
| 10 import os | |
| 11 import shutil | |
| 12 import sys | |
| 13 | |
| 14 try: | |
| 15 parser = argparse.ArgumentParser(allow_abbrev=False, description=__doc__) | |
| 16 except TypeError: | |
| 17 parser = argparse.ArgumentParser(description=__doc__) | |
| 18 | |
| 19 parser.add_argument( | |
| 20 'name', | |
| 21 help='The base name of the logfile to import', | |
| 22 metavar='LOG', | |
| 23 type=str, | |
| 24 ) | |
| 25 | |
| 26 parser.add_argument( | |
| 27 '-s', '--source', | |
| 28 help='The name (recommended) or IP of the source host', | |
| 29 metavar='HOSTNAME', | |
| 30 type=str, | |
| 31 ) | |
| 32 | |
| 33 parser.add_argument( | |
| 34 '-t', '--target', | |
| 35 help='The location of the upload/import directory', | |
| 36 metavar='DIRECTORY', | |
| 37 type=str, | |
| 38 ) | |
| 39 | |
| 40 arguments = parser.parse_args() | |
| 41 destination = os.path.join(arguments.target, arguments.source, arguments.name) | |
| 42 output = open(destination, 'a') | |
| 43 fcntl.flock(output, fcntl.F_WRLCK | fcntl.F_EXLCK) | |
| 44 | |
| 45 try: | |
| 46 shutil.copyfileobj(sys.stdin, output) | |
| 47 finally: | |
| 48 fcntl.flock(output, fcntl.F_UNLCK) | |
| 49 | |
| 50 output.close() | |
| OLD | NEW |