diff --git a/ihatemoney/models.py b/ihatemoney/models.py index 10615d42..212f8949 100644 --- a/ihatemoney/models.py +++ b/ihatemoney/models.py @@ -678,7 +678,8 @@ class Bill(db.Model): original_currency = db.Column(db.String(3)) converted_amount = db.Column(db.Float) - archive = db.Column(db.Integer, db.ForeignKey("archive.id")) + # archive = db.Column(db.Integer, db.ForeignKey("archive.id")) + archive = db.Column(db.Boolean) currency_helper = CurrencyConverter() @@ -692,6 +693,7 @@ class Bill(db.Model): payer_id: int = None, project_default_currency: str = "", what: str = "", + archive: bool = False ): super().__init__() self.amount = amount @@ -704,6 +706,7 @@ class Bill(db.Model): self.converted_amount = self.currency_helper.exchange_currency( self.amount, self.original_currency, project_default_currency ) + self.archive = archive @property def _to_serialize(self): @@ -750,6 +753,9 @@ class Bill(db.Model): f"{', '.join([o.name for o in self.owers])}>" ) + def set_archived(self): + self.archive = True + class Archive(db.Model): id = db.Column(db.Integer, primary_key=True)