| from google.cloud import bigquery |
| import functions_framework |
| from queries import queries |
| from google.oauth2 import service_account |
| import json |
| import requests |
| import streamlit as st |
| import pyperclip |
| from ap import send_message_via_webhook, Webhook_urls |
|
|
| |
| webhook_url = list(Webhook_urls.keys()) |
| html_subject = """ |
| <html> |
| <head> |
| <style> |
| .button { |
| display: inline-block; |
| padding: 10px 20px; |
| border-radius: 12px; |
| background: linear-gradient(to bottom, #f8f9fa, #e0e0e0); |
| box-shadow: |
| 0 6px 12px rgba(0, 0, 0, 0.3), |
| 0 8px 16px rgba(0, 0, 0, 0.2), |
| inset 0 -2px 4px rgba(255, 255, 255, 0.6); |
| text-align: center; |
| position: relative; |
| transform: translateY(4px); |
| transition: transform 0.2s ease-in-out, box-shadow 0.2s ease-in-out; |
| cursor: pointer; |
| user-select: none; |
| } |
| .button:hover { |
| box-shadow: |
| 0 8px 16px rgba(0, 0, 0, 0.3), |
| 0 12px 24px rgba(0, 0, 0, 0.2); |
| transform: translateY(2px); |
| } |
| .button:active { |
| box-shadow: |
| 0 4px 8px rgba(0, 0, 0, 0.3), |
| 0 6px 12px rgba(0, 0, 0, 0.2); |
| transform: translateY(0); |
| } |
| </style> |
| </head> |
| <body> |
| <div class="button"> |
| <h3 style=" |
| font-size: 20px; |
| color: #ffffff; |
| background-image: linear-gradient(to right, #800000, #ff0000, #ffdab9); |
| background-clip: text; |
| -webkit-background-clip: text; |
| text-fill-color: transparent; |
| -webkit-text-fill-color: transparent; |
| margin: 0; |
| text-shadow: 0 2px 5px rgba(0, 0, 0, 0.4); |
| ">Select channels/members</h3> |
| </div> |
| </body> |
| </html> |
| """ |
|
|
|
|
| st.markdown(html_subject, unsafe_allow_html=True) |
| selection = st.multiselect("", webhook_url) |
| |
|
|
|
|
|
|
| def check_duplicates(credentials_file): |
| """Check for duplicates using BigQuery with the provided credentials file.""" |
| results = {} |
| credentials = service_account.Credentials.from_service_account_info(json.loads(credentials_file)) |
| client = bigquery.Client(credentials=credentials, project=credentials.project_id) |
|
|
| for i, (query_name, query) in enumerate(queries.items()): |
| |
| query_job = client.query(query) |
| df = query_job.result().to_dataframe() |
| |
| st.write(f"{query_name}:", df) |
|
|
| button_styles = """ |
| <style> |
| div.stButton > button { |
| color: #ffffff; /* Text color */ |
| font-size: 30px; |
| background-image: linear-gradient(to right, #800000, #ff0000); /* Maroon to light red gradient */ |
| border: none; |
| padding: 10px 20px; |
| cursor: pointer; |
| border-radius: 15px; |
| display: inline-block; |
| box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1), 0 8px 15px rgba(0, 0, 0, 0.1); /* Box shadow */ |
| transition: all 0.3s ease; /* Smooth transition on hover */ |
| } |
| div.stButton > button:hover { |
| background-color: #00ff00; /* Hover background color */ |
| color: #ff0000; /* Hover text color */ |
| box-shadow: 0 6px 10px rgba(0, 0, 0, 0.2), 0 12px 20px rgba(0, 0, 0, 0.2); /* Box shadow on hover */ |
| } |
| </style> |
| """ |
| st.markdown(button_styles, unsafe_allow_html=True) |
| if st.button(f"Copy Query", key=f"copy_query_{i}"): |
| pyperclip.copy(query) |
| st.success('Query copied to clipboard!') |
|
|
| if not df.empty: |
| duplicate_count = len(df) |
| results[query_name] = duplicate_count |
|
|
| return results |
|
|
| |
| html_subject = """ |
| <html> |
| <head> |
| <style> |
| .button { |
| display: inline-block; |
| padding: 10px 20px; |
| border-radius: 12px; |
| background: linear-gradient(to bottom, #f8f9fa, #e0e0e0); |
| box-shadow: |
| 0 6px 12px rgba(0, 0, 0, 0.3), |
| 0 8px 16px rgba(0, 0, 0, 0.2), |
| inset 0 -2px 4px rgba(255, 255, 255, 0.6); |
| text-align: center; |
| position: relative; |
| transform: translateY(4px); |
| transition: transform 0.2s ease-in-out, box-shadow 0.2s ease-in-out; |
| cursor: pointer; |
| user-select: none; |
| } |
| .button:hover { |
| box-shadow: |
| 0 8px 16px rgba(0, 0, 0, 0.3), |
| 0 12px 24px rgba(0, 0, 0, 0.2); |
| transform: translateY(2px); |
| } |
| .button:active { |
| box-shadow: |
| 0 4px 8px rgba(0, 0, 0, 0.3), |
| 0 6px 12px rgba(0, 0, 0, 0.2); |
| transform: translateY(0); |
| } |
| </style> |
| </head> |
| <body> |
| <div class="button"> |
| <h3 style=" |
| font-size: 20px; |
| color: #ffffff; |
| background-image: linear-gradient(to right, #800000, #ff0000, #ffdab9); |
| background-clip: text; |
| -webkit-background-clip: text; |
| text-fill-color: transparent; |
| -webkit-text-fill-color: transparent; |
| margin: 0; |
| text-shadow: 0 2px 5px rgba(0, 0, 0, 0.4); |
| ">Upload service a/c credentials</h3> |
| </div> |
| </body> |
| </html> |
| """ |
|
|
| st.markdown(html_subject, unsafe_allow_html=True) |
|
|
| |
| credentials_file = st.file_uploader("", type="json") |
|
|
| if credentials_file is not None: |
| |
| credentials_data = credentials_file.read().decode("utf-8") |
| |
| |
| results = check_duplicates(credentials_data) |
| st.write("") |
| st.write("") |
| |
| if results: |
| |
| html_subject = """ |
| <html> |
| <head> |
| <style> |
| .button { |
| display: inline-block; |
| padding: 10px 20px; |
| border-radius: 12px; |
| background: linear-gradient(to bottom, #f8f9fa, #e0e0e0); |
| box-shadow: |
| 0 6px 12px rgba(0, 0, 0, 0.3), |
| 0 8px 16px rgba(0, 0, 0, 0.2), |
| inset 0 -2px 4px rgba(255, 255, 255, 0.6); |
| text-align: center; |
| position: relative; |
| transform: translateY(4px); |
| transition: transform 0.2s ease-in-out, box-shadow 0.2s ease-in-out; |
| cursor: pointer; |
| user-select: none; |
| } |
| .button:hover { |
| box-shadow: |
| 0 8px 16px rgba(0, 0, 0, 0.3), |
| 0 12px 24px rgba(0, 0, 0, 0.2); |
| transform: translateY(2px); |
| } |
| .button:active { |
| box-shadow: |
| 0 4px 8px rgba(0, 0, 0, 0.3), |
| 0 6px 12px rgba(0, 0, 0, 0.2); |
| transform: translateY(0); |
| } |
| .spacing { |
| margin: 20px 0; /* Adjust the value as needed */ |
| } |
| .result-box { |
| border: 1px solid #ddd; |
| border-radius: 8px; |
| background: #f9f9f9; |
| box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1); |
| padding: 10px; |
| margin-bottom: 10px; |
| font-size: 16px; |
| } |
| .result-title { |
| font-weight: bold; |
| color: #333; |
| } |
| </style> |
| </head> |
| <body> |
| <div class="button"> |
| <h3 style=" |
| font-size: 20px; |
| color: #ffffff; |
| background-image: linear-gradient(to right, #800000, #ff0000, #ffdab9); |
| background-clip: text; |
| -webkit-background-clip: text; |
| text-fill-color: transparent; |
| -webkit-text-fill-color: transparent; |
| margin: 0; |
| text-shadow: 0 2px 5px rgba(0, 0, 0, 0.4); |
| ">Duplicate Counts for Queries:</h3> |
| </div> |
| <div class="spacing"></div> <!-- Adds spacing between the heading and the content --> |
| </body> |
| </html> |
| """ |
|
|
| |
| st.markdown(html_subject, unsafe_allow_html=True) |
|
|
| |
| message_html = "" |
| for query_name, count in results.items(): |
| message_html += f""" |
| <div class="result-box"> |
| <div class="result-title">{query_name}</div> |
| <div>{count} duplicate rows</div> |
| </div> |
| """ |
|
|
| |
| st.markdown(message_html, unsafe_allow_html=True) |
|
|
| |
| message_text = "" |
| for query_name, count in results.items(): |
| message_text += f"*{query_name}*\n{count} duplicate rows\n\n" |
|
|
| if not results: |
| message_text = "No duplicates found in the queries." |
|
|
| |
| for channel in selection: |
| webhook_url = Webhook_urls.get(channel) |
| if webhook_url: |
| send_message_via_webhook(message_text, webhook_url) |
| else: |
| st.error(f"Webhook URL not found for channel: {channel}") |
