Refactor get_cnames function to accept wildcard parameter

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 from generate_cnames import generate_cnames
 
-def get_cnames(i_domains):
+def get_cnames(i_domains, wildcard=False):
     cf_domain = "silerudaagartha.eu.org"
-    cname_recs, cname_values = generate_cnames(i_domains, cf_domain)
+    if wildcard:
+        cname_recs, cname_values = generate_cnames(i_domains, cf_domain, wildcard)
+    else:
+        cname_recs, cname_values = generate_cnames(i_domains, cf_domain, wildcard)
     table_data = []
     for rec, value in zip(cname_recs, cname_values):
         table_data.append([rec, value])
@@ -13,10 +16,11 @@ def app():
     with gr.Blocks(title="Generate CNAMES for Project Gatekeeper") as webui:
         with gr.Row():
             cname_domains = gr.Textbox(label="Enter Domains", placeholder="thenayankasturi.eu.org, dash.thenayankasturi.eu.org, www.thenayankasturi.eu.org", type="text", interactive=True)
+            wildcard = gr.Checkbox(label="Wildcard", value=False)
             btn = gr.Button(value="Generate CNAME Records & Values")
         with gr.Row():
             records = gr.Dataframe(label="CNAME Records", headers=["CNAME", "CNAME VALUE"], row_count=(1), col_count=(2))
-        btn.click(get_cnames, inputs=cname_domains, outputs=records)
+        btn.click(get_cnames, inputs=[cname_domains, wildcard], outputs=records)
     try:
         webui.queue(default_concurrency_limit=25).launch()
     except Exception as e:

generate_cnames.py

@@ -36,12 +36,26 @@ def gen_cname_values(domains, cf_domain, exchange):
     return cname_values
 
 def gen_cname(domains, cf_domain, exchange):
+    print(domains)
     cname_recs  = gen_cname_recs(domains)
     cname_values = gen_cname_values(domains, cf_domain, exchange)
     return cname_recs, cname_values
 
-def generate_cnames(i_domains, cf_domain):
+def generate_cnames(i_domains, cf_domain, wildcard=False):
     domains = get_domains(i_domains)
     exchange = extract_subdomains(domains=domains)
-    cname_recs, cname_values = gen_cname(domains, cf_domain, exchange)
-    return cname_recs, cname_values
+    if wildcard:
+        domain = []
+        domain.append(f"{exchange}")
+        cname_recs, cname_values = gen_cname(domain, cf_domain, exchange)
+    else:
+        cname_recs, cname_values = gen_cname(domains, cf_domain, exchange)
+    return cname_recs, cname_values
+
+# i_domains = "dash.raanna.com, raanna.com"
+# cf_domain = "silerudaagartha.eu.org"
+# wildcard = True
+# cname_recs, cname_values = generate_cnames(i_domains, cf_domain, wildcard)
+# for rec, value in zip(cname_recs, cname_values):
+#     print(f"{rec} -> {value}")
+