Add field to set a project's default branch

Author: tobami

codespeed/migrations/0003_project_default_branch.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.9.13 on 2017-08-04 03:45
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+def get_default_branch_name():
+    from django.conf import settings
+    try:
+        return settings.DEF_BRANCH
+    except AttributeError:
+        return "master"
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('codespeed', '0002_median'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='project',
+            name='default_branch',
+            field=models.CharField(default=get_default_branch_name, max_length=32),
+            preserve_default=False,
+        ),
+        migrations.AlterField(
+            model_name='branch',
+            name='name',
+            field=models.CharField(max_length=32),
+        ),
+    ]

codespeed/models.py

@@ -42,6 +42,7 @@ class Project(models.Model):
     commit_browsing_url = models.CharField("Commit browsing URL",
                                            blank=True, max_length=200)
     track = models.BooleanField("Track changes", default=True)
+    default_branch = models.CharField(max_length=32)
 
     def __str__(self):
         return self.name
@@ -103,7 +104,7 @@ def is_less_important_than(self, val, color):
 
 @python_2_unicode_compatible
 class Branch(models.Model):
-    name = models.CharField(max_length=20)
+    name = models.CharField(max_length=32)
     project = models.ForeignKey(Project, related_name="branches")
 
     def __str__(self):