From 4ce5d78b23d7c3e93a5624cf753142b236d9f720 Mon Sep 17 00:00:00 2001
From: Marc <marcvila10@hotmail.com>
Date: Tue, 8 Jan 2019 07:38:59 +0100
Subject: [PATCH] Adapt to new model (WIP) - 2

---
 script/country.py   | 2 +-
 script/my_script.py | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/script/country.py b/script/country.py
index d5eb6dd6..fd35617d 100644
--- a/script/country.py
+++ b/script/country.py
@@ -8,7 +8,7 @@ class Country:
 
     def get_ambit(self, ambit_to_get):
         for ambit in self.ambits:
-            if ambit.name == ambit_to_get.name:
+            if ambit.name == ambit_to_get:
                 return ambit
         return None
 
diff --git a/script/my_script.py b/script/my_script.py
index 237a0648..50a77994 100644
--- a/script/my_script.py
+++ b/script/my_script.py
@@ -65,6 +65,7 @@ def get_channels_from_part(text):
     return channel_list
 
 
+# TODO Change this
 page = requests.get('https://raw.githubusercontent.com/LaQuay/TDTChannels/add-local-m3u8/TELEVISION.md',
                     headers={'Cache-Control': 'no-cache'})
 content = str(page.text)
@@ -144,7 +145,7 @@ spain.add_ambit(Ambito("Navarra", get_channels_from_part(canales_autonomicos_nav
 canales_autonomicos_pais_vasco = stringbetween(content_nacional, "#### País Vasco", "#### Valencia")
 spain.add_ambit(Ambito("País Vasco", get_channels_from_part(canales_autonomicos_pais_vasco)))
 
-canales_autonomicos_valencia = stringbetween(content_nacional, "#### Valencia", "### Locales")
+canales_autonomicos_valencia = stringbetween(content_nacional, "#### Valencia", "")
 spain.add_ambit(Ambito("Valencia", get_channels_from_part(canales_autonomicos_valencia)))
 
 canales_locales_andalucia = stringbetween(content_local, "#### Andalucía", "#### Aragón")
@@ -201,7 +202,7 @@ spain.get_ambit("Navarra").add_channels(get_channels_from_part(canales_locales_n
 canales_locales_pais_vasco = stringbetween(content_local, "#### País Vasco", "#### Valencia")
 spain.get_ambit("País Vasco").add_channels(get_channels_from_part(canales_locales_pais_vasco))
 
-canales_locales_valencia = stringbetween(content_local, "#### Valencia", "## Locales")
+canales_locales_valencia = stringbetween(content_local, "#### Valencia", "")
 spain.get_ambit("Valencia").add_channels(get_channels_from_part(canales_locales_valencia))
 
 canales_internacionales = stringbetween(content, "## Internacionales", "### Andorra")