Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

support reading from stdin in bin/bitbot-log

Browse source
This commit is contained in:
jesopo 2020-02-24 17:58:00 +00:00
parent 36fd91f009
commit bb975f8e60
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
bin/bitbot-log
View file

@ -11,7 +11,7 @@ arg_parser.add_argument("log", help="Location of the log file to decrypt")
args = arg_parser.parse_args() args = arg_parser.parse_args()
import base64 import base64, sys
from cryptography.hazmat.backends import default_backend from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives import serialization from cryptography.hazmat.primitives import serialization
from cryptography.hazmat.primitives import hashes from cryptography.hazmat.primitives import hashes
@ -38,10 +38,14 @@ def aes_decrypt(key: bytes, data_str: str):
with open(args.key, "rb") as key_file: with open(args.key, "rb") as key_file:
key_content = key_file.read() key_content = key_file.read()
key = serialization.load_pem_private_key( key = serialization.load_pem_private_key(
key_content, password=None, backend=default_backend()) key_content, password=None, backend=default_backend())
with open(args.log) as log_file: if args.log == "-":
lines = sys.stdin.read().split("\n")
else:
with open(args.log) as log_file:
lines = log_file.read().split("\n") lines = log_file.read().split("\n")
lines = filter(None, lines) lines = filter(None, lines)