Add constraint to limit maximum battery expansion in Germany

etrago/tools/constraints.py

@@ -189,6 +189,44 @@ def _rule(m):
 
     network.model.max_line_ext = Constraint(rule=_rule)
 
+
+def _max_battery_expansion_germany(self, network, snapshots):
+    """
+    Set maximum expanded capacity of batteries in Germany.
+
+    Parameters
+    ----------
+    network : :class:`pypsa.Network
+        Overall container of PyPSA
+    snapshots : pandas.DatetimeIndex
+        List of timesteps considered in the optimization
+
+    Returns
+    -------
+    None.
+
+    """
+    home_battery_capacity = network.storage_units[
+        network.storage_units.carrier=="battery"].p_nom_min.sum()
+
+    batteries = network.storage_units[(
+        network.storage_units.carrier=="battery")
+        &(network.storage_units.bus.isin(
+            network.buses[network.buses.country=="DE"].index))]
+
+    def _rule_max(m):
+        batteries_opt = sum(
+            m.storage_p_nom[index]
+            for index in batteries.index
+        )
+        return batteries_opt <= (
+            1
+        ) * (self.args["extra_functionality"]["max_battery_expansion_germany"]
+            + home_battery_capacity)
+
+    network.model.max_battery_ext = Constraint(rule=_rule_max)
+
+
 def _fixed_battery_expansion_germany(self, network, snapshots):
     """
     Define the overall expanded capacity of batteries in Germany.