Allow adding domains to an app that does not have domains yet.

ojii · Apr 11, 2012 · f1605ad · f1605ad
1 parent 9b2bf11
commit f1605ad
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/heroku/api.py b/heroku/api.py
@@ -120,6 +120,7 @@ def _get_resources(self, resource, obj, params=None, map=None, **kwargs):
         list_resource = map(items=items)
         list_resource._h = self
         list_resource._obj = obj
+        list_resource._kwargs = kwargs
 
         return list_resource
 

diff --git a/heroku/structures.py b/heroku/structures.py
@@ -17,6 +17,7 @@ def __init__(self, items=None):
         self._h = None
         self._items = items or list()
         self._obj = None
+        self._kwargs = {}
 
     def __repr__(self):
         return repr(self._items)
@@ -40,14 +41,17 @@ def __getitem__(self, key):
         return v
 
     def add(self, *args, **kwargs):
+        full_kwargs = {}
+        full_kwargs.update(self._kwargs)
+        full_kwargs.update(kwargs)
 
         try:
-            return self[0].new(*args, **kwargs)
+            return self[0].new(*args, **full_kwargs)
         except IndexError:
             o = self._obj()
             o._h = self._h
 
-            return o.new(*args, **kwargs)
+            return o.new(*args, **full_kwargs)
 
 
     def remove(self, key):