diff options
Diffstat (limited to 'html_builders/elements.py')
| -rw-r--r-- | html_builders/elements.py | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/html_builders/elements.py b/html_builders/elements.py new file mode 100644 index 0000000..41c8525 --- /dev/null +++ b/html_builders/elements.py @@ -0,0 +1,91 @@ +import math +import pandas as pd +import warnings +from datetime import datetime, timedelta + +def build_title_banner(text: str, menu_items): + banner_html = f""" + <div class="banner-container"> + <div class="banner-text">{text}</div> + </div> + <div class="menu-bar"> + """ + for item in menu_items: + banner_html += f""" + <a href="{item[1]}">{item[0]}</a> + + """ + banner_html += "</div>" + return banner_html + +def build_projection_card(server, table_name, curr_subscribers, channel_name, + timezone = "Pacific Standard Time"): + def get_next_milestone(subscriber_count): + num_digits = len(str(subscriber_count)) + if num_digits <= 4: + milestone_interval = 10000 + else: + milestone_interval = 10 ** (num_digits - 1) + next_milestone = math.ceil( + subscriber_count / milestone_interval) * milestone_interval + + return next_milestone + + warnings.filterwarnings('ignore') # Ignore pandas warning regarding pyodbc + def create_milestone_card(time_until_milestone, next_milestone, not_enough_data = False, + declining = False): + now = datetime.now() + milestone_date = ( + now + timedelta(seconds = time_until_milestone)).strftime('%Y-%m-%d') + relative_time = now + timedelta(seconds = time_until_milestone) - now + next_milestone_str = "{:,}".format(next_milestone) + if relative_time.days > 1: + relative_time_str = f"In {relative_time.days} days" + elif relative_time.days == 1: + relative_time_str = f"In {relative_time.days} day" + elif relative_time.days < 0: + relative_time_str = f"{-relative_time.days} days ago" + elif not_enough_data: + relative_time_str = "Not enough data" + elif declining: + relative_time_str = "Declining" + else: + relative_time_str = "Today" + card = f""" + <div style="background-color: #ffffff; padding: 20px; border-radius: 10px; box-shadow: 0 0 10px rgba(0, 0, 0, 0.2); font-family: 'Poppins', sans-serif;"> + <div style="font-size: 24px; font-weight: bold; margin-bottom: 10px;">Next Milestone</div> + <div style="font-size: 18px; margin-bottom: 10px;">{next_milestone_str}</div> + <div style="display: flex; flex-direction: row; justify-content: space-between; margin-bottom: 10px;"> + <div style="font-size: 16px;">Estimated Date:</div> + <div style="font-size: 16px; font-weight: bold;">{milestone_date}</div> + </div> + <div style="display: flex; flex-direction: row; justify-content: space-between; color: #888;"> + <div style="font-size: 16px;">{relative_time_str}</div> + <div style="font-size: 16px;">{timezone}</div> + </div> + </div> + <br> + """ + return card + query = f"SELECT name, subscriber_count, timestamp FROM {table_name} WHERE timestamp >= DATE_SUB(CURRENT_DATE(), INTERVAL 30 DAY) AND name = \"{channel_name}\" ORDER BY timestamp DESC" + df = pd.read_sql_query(query, server.get_connection()) + df = df.sort_values(by = 'timestamp') + # get the rate of change from first data point to last data point + first_data = df.iloc[0] + last_data = df.iloc[-1] + delta_sub_count = last_data['subscriber_count'] - first_data['subscriber_count'] + delta_time = (last_data['timestamp'] - first_data['timestamp']).total_seconds() + + # Calculate the average rate of change of subscriber_count over time + avg_rate_of_change = delta_sub_count / delta_time + next_milestone = get_next_milestone(curr_subscribers) + if avg_rate_of_change == 0 or math.isnan(avg_rate_of_change) or math.isinf(avg_rate_of_change): + return create_milestone_card(0, next_milestone, not_enough_data = True) + if avg_rate_of_change < 0: + return create_milestone_card(0, next_milestone, declining = True) + time_to_next_milestone = ( + next_milestone - curr_subscribers) / avg_rate_of_change + return create_milestone_card(time_to_next_milestone, next_milestone) + + + |