refactoring

src/CountryOptionsSetter.py

@@ -0,0 +1,21 @@
+from bs4 import BeautifulSoup
+
+
+class CountryOptionsSetter:
+
+    def setCountryOptions(self, html, options):
+        soup = self._setCountryOptions(self._parse(html), self._parseOptions(options))
+        return str(soup)
+
+    def _setCountryOptions(self, soup, options):
+        countrySelect = soup.find(id = "countrySelect")
+        countrySelect.clear()
+        for option in options:
+            countrySelect.append(option)
+        return soup
+
+    def _parseOptions(self, options):
+        return [self._parse(option).option for option in options]
+
+    def _parse(self, html):
+        return BeautifulSoup(html, 'lxml')