From b9664263d01eb97585d6c6fe8240b7ab1db49eed Mon Sep 17 00:00:00 2001
From: George Lacey <git@george.ooo>
Date: Tue, 4 May 2021 12:04:46 +0100
Subject: [PATCH] Add primary key attribute to borg classes

---
 src/borg/archive.py | 3 ++-
 src/borg/error.py   | 3 ++-
 src/borg/repo.py    | 3 ++-
 src/borg/stats.py   | 3 ++-
 4 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/borg/archive.py b/src/borg/archive.py
index 3a3ec1b..ec8a7ed 100644
--- a/src/borg/archive.py
+++ b/src/borg/archive.py
@@ -2,11 +2,12 @@ from datetime import datetime
 
 
 class Archive(object):
-    def __init__(self, fingerprint: str, name: str, start: datetime, end: datetime):
+    def __init__(self, fingerprint: str, name: str, start: datetime, end: datetime, primary_key=None):
         self.fingerprint = fingerprint
         self.name = name
         self.start = start
         self.end = end
+        self.primary_key = primary_key
 
     @classmethod
     def from_json(cls, json: dict):
diff --git a/src/borg/error.py b/src/borg/error.py
index 774f94b..de1595a 100644
--- a/src/borg/error.py
+++ b/src/borg/error.py
@@ -2,9 +2,10 @@ from datetime import datetime
 
 
 class Error(object):
-    def __init__(self, error: str, time: datetime):
+    def __init__(self, error: str, time: datetime, primary_key=None):
         self.error = error
         self.time = time
+        self.primary_key = primary_key
 
     @classmethod
     def from_json(cls, json: dict):
diff --git a/src/borg/repo.py b/src/borg/repo.py
index 7187378..3655ca1 100644
--- a/src/borg/repo.py
+++ b/src/borg/repo.py
@@ -3,10 +3,11 @@ from pathlib import Path
 
 
 class Repo(object):
-    def __init__(self, fingerprint: str, location: Path, last_modified: datetime):
+    def __init__(self, fingerprint: str, location: Path, last_modified: datetime, primary_key=None):
         self.fingerprint = fingerprint
         self.location = location
         self.last_modified = last_modified
+        self.primary_key = primary_key
 
     @classmethod
     def from_json(cls, json: dict):
diff --git a/src/borg/stats.py b/src/borg/stats.py
index e4e19d4..d847521 100644
--- a/src/borg/stats.py
+++ b/src/borg/stats.py
@@ -1,9 +1,10 @@
 class Stats(object):
-    def __init__(self, file_count: int, original_size: int, compressed_size: int, deduplicated_size: int):
+    def __init__(self, file_count: int, original_size: int, compressed_size: int, deduplicated_size: int, primary_key=None):
         self.file_count = file_count
         self.original_size = original_size
         self.compressed_size = compressed_size
         self.deduplicated_size = deduplicated_size
+        self.primary_key = primary_key
 
     @classmethod
     def from_json(cls, json: dict):