Add settlements create command

piket_client/cli.py

@@ -56,7 +56,24 @@ def show_settlement(settlement_id: int) -> None:
         print_error(f"Could not get Settlement: {s.value}")
         return
 
-    click.echo(f"Settlement {settlement_id}, \"{s.name}\"")
+    output_settlement_info(s)
+
+
+@settlements.command("create")
+@click.argument("name")
+def create_settlement(name: str) -> None:
+    """Create a new Settlement."""
+    s = Settlement.create(name)
+
+    if isinstance(s, NetworkError):
+        print_error(f"Could not create Settlement: {s.value}")
+        return
+
+    output_settlement_info(s)
+
+
+def output_settlement_info(s: Settlement) -> None:
+    click.echo(f'Settlement {s.settlement_id}, "{s.name}"')
 
     click.echo(f"Summary:")
     for key, value in s.consumption_summary.items():

piket_server/models.py

@@ -89,6 +89,7 @@ class Settlement(db.Model):
             "consumption_summary": self.consumption_summary,
             "unique_people": self.unique_people,
             "per_person_counts": self.per_person_counts,
+            "count_info": self.per_person,
         }
 
     @property

piket_server/routes/settlements.py

@@ -46,4 +46,4 @@ def add_settlement():
 
     db.session.commit()
 
-    return jsonify(settlement=s.as_dict)
+    return jsonify(settlement=s.as_dict, count_info=s.per_person)