From 76425437960f1cd56df4f0bd342e7cc9c788bd8c Mon Sep 17 00:00:00 2001
From: Yuuki <yuuki@yuuki-chan.xyz>
Date: Thu, 2 Jan 2025 21:11:15 +0900
Subject: [PATCH] Add 5.1 audio attribute support.

---
 constants.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/constants.py b/constants.py
index 981a1e5..ca38957 100644
--- a/constants.py
+++ b/constants.py
@@ -30,11 +30,13 @@ class XMLTV:
             open('data/channels.xml', mode='a', encoding='UTF-8').write(c)
 
     def add_programme(self, channel, title, desc, start, stop, category, attr, icon):
+        # TODO: add `duo` attribute (bilingual)
         programme = '<programme start="{} +0000" stop="{} +0000" channel="{}">'.format(start, stop, channel)
         programme += '<title>{}</title>'.format(title)
         programme += '<desc>{}</desc>'.format(desc)
         programme += '<category lang="ja">{}</category>'.format(category)
         if icon: programme += '<icon src="{}" />'.format(icon)
+        if '5.1' in attr: programme += '<audio><stereo>surround</stereo></audio>'
         if 'stereo' in attr: programme += '<audio><stereo>stereo</stereo></audio>'
         if 'mono' in attr: programme += '<audio><stereo>mono</stereo></audio>'
         if 'hd' in attr: programme += '<video><present>yes</present><aspect>16:9</aspect><quality>HDTV</quality></video>'