From 9c03c9bdc55a78c9dac3289954273ea046d88853 Mon Sep 17 00:00:00 2001
From: Daan Sprenkels <hello@dsprenkels.com>
Date: Sat, 16 Sep 2017 13:15:35 +0200
Subject: [PATCH] replaygain: Prepare song model for rg retrieval

---
 marietje/marietje/utils.py                   |  4 ++-
 marietje/songs/migrations/0002_replaygain.py | 30 ++++++++++++++++++++
 marietje/songs/models.py                     |  6 +++-
 3 files changed, 38 insertions(+), 2 deletions(-)
 create mode 100644 marietje/songs/migrations/0002_replaygain.py

diff --git a/marietje/marietje/utils.py b/marietje/marietje/utils.py
index a0dc2a7..9942173 100644
--- a/marietje/marietje/utils.py
+++ b/marietje/marietje/utils.py
@@ -8,7 +8,9 @@ def song_to_dict(song, hash=False, user=False):
         'id': song.id,
         'artist': song.artist,
         'title': song.title,
-        'duration': song.duration
+        'duration': song.duration,
+        'rg_gain': song.rg_gain,
+        'rg_peak': song.rg_peak,
     }
 
     if hash:
diff --git a/marietje/songs/migrations/0002_replaygain.py b/marietje/songs/migrations/0002_replaygain.py
new file mode 100644
index 0000000..c7a3f5d
--- /dev/null
+++ b/marietje/songs/migrations/0002_replaygain.py
@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.5 on 2017-09-18 12:14
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('songs', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='song',
+            name='rg_gain',
+            field=models.DecimalField(blank=True, decimal_places=6, help_text='replaygain gain level', max_digits=9, null=True),
+        ),
+        migrations.AddField(
+            model_name='song',
+            name='rg_peak',
+            field=models.DecimalField(blank=True, decimal_places=6, help_text='replaygain peak level', max_digits=9, null=True),
+        ),
+        migrations.AlterField(
+            model_name='song',
+            name='duration',
+            field=models.IntegerField(help_text='track duration in seconds'),
+        ),
+    ]
diff --git a/marietje/songs/models.py b/marietje/songs/models.py
index f0e23d3..0c25de7 100644
--- a/marietje/songs/models.py
+++ b/marietje/songs/models.py
@@ -12,7 +12,11 @@ class Song(models.Model):
     artist = models.TextField()
     title = models.TextField()
     hash = models.TextField()
-    duration = models.IntegerField()
+    duration = models.IntegerField(help_text="track duration in seconds")
+    rg_gain = models.DecimalField(max_digits=9, decimal_places=6,
+        blank=True, null=True, help_text="replaygain gain level")
+    rg_peak = models.DecimalField(max_digits=9, decimal_places=6,
+        blank=True, null=True, help_text="replaygain peak level")
     old_id = models.TextField(blank=True, null=True, default=None)
     deleted = models.BooleanField(default=False)