Tugas Pemrograman Ekstra - Soal 2 Tahap 1 - 1806196806

TPEkstra_2_1806196806.py

@@ -13,21 +13,17 @@ def get_data(situs):
 
 def get_data_per_wilayah(nama_wilayah):
     text = get_data("https://pudyprima.github.io/covid19-jakarta").lower()
+    nama_wilayah = nama_wilayah.lower()
+    singkatan = ['jaksel','jakpus','jakbar','jaktim','jakut']
     if text == "terjadi eror":
         return "Terjadi eror, pastikan internet anda stabil"
     else:
-        if nama_wilayah[0:3].lower() == 'jak':
-            if nama_wilayah[3:].lower() == 'sel':
-                nama_wilayah='jakarta selatan'
-            elif nama_wilayah[3:].lower() == 'bar':
-                nama_wilayah='jakarta barat'
-            elif nama_wilayah[3:].lower() == 'tim':
-                nama_wilayah='jakarta timur'
-            elif nama_wilayah[3:].lower() == 'pus':
-                nama_wilayah='jakarta pusat'
-            elif nama_wilayah[3:].lower() == 'ut' :
-                nama_wilayah='jakarta utara'
-        mulai = text.find("<td>"+nama_wilayah.lower()+"</td>")
+        cari_singkatan = nama_wilayah in singkatan
+        if cari_singkatan == True:
+            index = singkatan.index(nama_wilayah)
+            singkatan = ['jakarta selatan','jakarta pusat', 'jakarta barat', 'jakarta timur', 'jakarta utara']
+            nama_wilayah = singkatan[index]
+        mulai = text.find("<td>"+nama_wilayah+"</td>")
         if mulai == -1:
             return "Maaf, "+nama_wilayah+" tidak ada dalam daftar wilayah."
         else:
@@ -68,10 +64,10 @@ def get_total_kasus_by_keyword(keyword):
             return jumlah
 
 def main():
-    print(get_data_per_wilayah('Jakarta barat'))
-    print(get_data_per_wilayah('Jakpus'))
+    print(get_data_per_wilayah('Jakbar'))
+    print(get_data_per_wilayah('Jakarta pusat'))
     print(get_data_per_wilayah('Jaksel'))
-    print(get_data_per_wilayah('Jaktim'))
+    print(get_data_per_wilayah('Jakarta timur'))
     print(get_data_per_wilayah('Jakut'))
     print(get_data_jakarta())
     print(get_total_kasus_by_keyword('perempuan'))