Add delete_branch_on_merge to orgs and main interface

nwalters512 · nwalters512 · commit 80009e464cfd · 2023-04-09T19:22:21.000-07:00
diff --git a/src/github3/github.py b/src/github3/github.py
@@ -788,6 +788,7 @@ def create_repository(
         auto_init=False,
         gitignore_template="",
         has_projects=True,
+        delete_branch_on_merge=False,
     ):
         """Create a repository for the authenticated user.
 
@@ -816,6 +817,9 @@ def create_repository(
         :param bool has_projects:
             (optional), If ``True``, enable projects for this repository. API
             default: ``True``
+        :param bool delete_branch_on_merge:
+            (optional) If ``True``, head branches will automatically be deleted
+            when pull requests are merged. API default: ``False``
         :returns:
             created repository
         :rtype:
@@ -832,6 +836,7 @@ def create_repository(
             "auto_init": auto_init,
             "gitignore_template": gitignore_template,
             "has_projects": has_projects,
+            "delete_branch_on_merge": delete_branch_on_merge,
         }
         json = self._json(self._post(url, data=data), 201)
         return self._instance_or_null(repo.Repository, json)
diff --git a/src/github3/orgs.py b/src/github3/orgs.py
@@ -623,6 +623,7 @@ def create_repository(
         gitignore_template="",
         license_template="",
         has_projects=True,
+        delete_branch_on_merge=False,
     ):
         """Create a repository for this organization.
 
@@ -662,6 +663,9 @@ def create_repository(
         :param bool has_projects:
             (optional), If ``True``, enable projects for this repository. API
             default: ``True``
+        :param bool delete_branch_on_merge:
+            (optional) If ``True``, head branches will automatically be deleted
+            when pull requests are merged. API default: ``False``
         :returns:
             the created repository
         :rtype:
@@ -679,6 +683,7 @@ def create_repository(
             "auto_init": auto_init,
             "gitignore_template": gitignore_template,
             "has_projects": has_projects,
+            "delete_branch_on_merge": delete_branch_on_merge,
         }
         if int(team_id) > 0:
             data.update({"team_id": team_id})
