From 8cb52d11507f81f7bbc466a1c18e5b6254a87040 Mon Sep 17 00:00:00 2001 From: George Lacey Date: Tue, 20 Sep 2016 15:05:12 +0100 Subject: [PATCH] Convert borg output to log_entry class - Remove old print code - Print using class method instead --- main.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/main.py b/main.py index dded17e..3664e93 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,7 @@ from sys import stdin +from log_entry import * -borg_output = stdin.readlines() +raw_borg_output = stdin.readlines() attributes = { "Archive name: " : "", "Archive fingerprint: " : "", @@ -8,17 +9,18 @@ attributes = { "Archive name: " : "", "Duration: " : "", "Number of files: " : "", } -for i in range(0, len(borg_output)): +for i in range(0, len(raw_borg_output)): for key in attributes: - if borg_output[i].startswith(key): - attributes[key] = borg_output[i] \ + if raw_borg_output[i].startswith(key): + attributes[key] = raw_borg_output[i] \ .strip(key) \ .rstrip() -output = open("borg.txt", "w") +borg_output = log_entry(attributes["Archive name: "], + attributes["Archive fingerprint: "], + attributes["Time (start): "], + attributes["Duration: "], + attributes["Number of files: "]) -for key, value in attributes.iteritems(): - output.write("%s%s\n" % (key, value)) - -output.close() +borg_output.print_to_file("borg.txt")