From 7bf8e284b95b258d040b22d2b1c969c41cd0aa0d Mon Sep 17 00:00:00 2001
From: Vivek Anand <vivekanand1101@gmail.com>
Date: Sep 30 2016 08:01:06 +0000
Subject: Adjust update_tickets_from_git to add milestones for issues as well


---

diff --git a/pagure/lib/git.py b/pagure/lib/git.py
index 4435e6a..0317bb8 100644
--- a/pagure/lib/git.py
+++ b/pagure/lib/git.py
@@ -470,6 +470,29 @@ def update_ticket_from_git(
 
     issue = pagure.lib.get_issue_by_uid(session, issue_uid=issue_uid)
 
+    # Update milestone
+    milestone = json_data.get('milestone')
+
+    # If milestone is not in the repo settings, add it
+    if milestone:
+        if milestone.strip() not in repo.milestones:
+            try:
+                repo.milestones[milestone.strip()] = None
+                session.add(repo)
+                session.commit()
+            except SQLAlchemyError:
+                session.rollback()
+    try:
+        msg = pagure.lib.edit_issue(
+            session,
+            issue=issue,
+            ticketfolder=None,
+            user=user.username,
+            milestone=milestone,
+        )
+    except SQLAlchemyError:
+        session.rollback()
+
     # Update tags
     tags = json_data.get('tags', [])
     pagure.lib.update_tags(