summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authornavanchauhan <navanchauhan@gmail.com>2023-05-30 11:03:26 -0600
committernavanchauhan <navanchauhan@gmail.com>2023-05-30 11:03:26 -0600
commite1dcdfef95738774683bbed187b0f3fdeffafd27 (patch)
tree40de885d6b6283f2945475af5415ed71a1081831
parentb4d5ed6b8b8112d817a089a68aa6f8fdb70de480 (diff)
fix dumbfuckery:
-rw-r--r--main.py569
1 files changed, 307 insertions, 262 deletions
diff --git a/main.py b/main.py
index 0369861..b5a0f15 100644
--- a/main.py
+++ b/main.py
@@ -12,6 +12,10 @@ import darkdetect
import os
from os import path
+from docx import Document
+from docx.shared import Inches
+from docx.enum.text import WD_ALIGN_PARAGRAPH
+
canadian_companies = []
def pyramid_list(lst, sort_canadian=True):
@@ -198,301 +202,342 @@ class CompanySelector:
def create_pyramid(self):
selected_companies_tmp = [self.company_listbox.listbox.get(index) for index in self.company_listbox.listbox.curselection()]
+ print(f'Selected {len(selected_companies_tmp)} companies')
selected_companies = [x.split(" - ")[0].strip() for x in selected_companies_tmp]
+ print(f'Symbols: {len(selected_companies)}')
selected_df = self.df[self.df["Symbol"].isin(selected_companies)].sort_values(by="Weighting", ascending=False)
+ print(f'DF: {len(selected_df)}')
#pyramid_file_path = asksaveasfilename(defaultextension=".xlsx", filetypes=[("Excel files", "*.xlsx *.xls")])
+ if self.generate_word_doc.get() == 1:
+ word_file_path = asksaveasfilename(defaultextension=".docx", filetypes=[("Word files", "*.docx")])
+ if not word_file_path:
+ return
- image_file_path = asksaveasfilename(defaultextension=".png", filetypes=[("Image files","*.png")])
-
- company_scores = {}
-
- # For each 'Company Name', company_scores[company] = 'Weighting' for that company
- for company, ticker, weighting, dividend, currency in zip(selected_df["Company Name"], selected_df["Symbol"], selected_df["Weighting"], selected_df["Dividend?"], selected_df["Currency"]):
- extra_char = ""
- if dividend.strip() == "Y":
- extra_char = "*"
- if currency == "CAD":
- canadian_companies.append(f'{company} ({ticker}){extra_char}')
- company_scores[f'{company} ({ticker}){extra_char}'] = weighting
-
- from PIL import Image, ImageDraw, ImageFont
- import math
- # Sort the dictionary by score in ascending order
- company_scores = {k: v for k, v in sorted(company_scores.items(), key=lambda item: item[1])}
-
- # Group the companies by their scores
- pyramid = {i: [] for i in range(1, 7)}
- for company, score in company_scores.items():
- pyramid[score].append(company)
-
- # Initialize some parameters
- img_width = 3300
- pyramid_width = 3000
- img_height = 1550 #2550
- overall_height = 2550
- block_color = (3, 37, 126) # Blue color
- dividend_color = block_color #(1,50,32) # Green color
- font_color = (255, 255, 255) # White color
- normal_block = (3, 37, 126) # Dark Blue
- padding = 15 # Padding around blocks
- radius = 10
- logo_padding = 100 # top padding for logo
- pyramid_padding_bottom = 350 # bottom padding for pyramid
- max_companies_in_row = 5
- base_path = "assets"
- primary_font = path.join(base_path, "baskerville.ttf")
- secondary_font = path.join(base_path, "gill_sans_bold.ttf")
-
- # Group the companies by their scores
- pyramid = {}
- for company, score in company_scores.items():
- if score not in pyramid:
- pyramid[score] = []
- pyramid[score].append(company)
-
- # Sort the pyramid keys in ascending order to have a pyramid shape
- sorted_keys = sorted(pyramid.keys())
-
- companies = []
-
-
- for key in sorted_keys:
- if pyramid[key] == []:
- continue
- if len(pyramid[key])<=max_companies_in_row:
- companies.append(pyramid[key])
- else:
- temp_list = []
- for x in range(len(pyramid[key])//max_companies_in_row):
- temp_list.append(pyramid[key][x*max_companies_in_row:max_companies_in_row])
-
- temp_list.append(pyramid[key][::-1][:len(pyramid[key])%max_companies_in_row])
- for row in temp_list[::-1]:
- if row != []:
- companies.append(row)
-
- everything_list = []
- for row in companies:
- if row == []:
- continue
- for company in row:
- everything_list.append(company)
-
- companies = everything_list
-
- companies = pyramid_list(companies)
-
- for company_row in companies:
- print(company_row)
-
- # Calculate the maximum number of companies in a group (this will be the width of your pyramid)
- max_companies = max(len(row) for row in companies) #max(len(v) for v in pyramid.values())
-
- # Calculate the total number of groups (the height of your pyramid)
- total_groups = len(companies) #len(pyramid)
-
- # Calculate the size of each block based on the width and height of the image and the number of blocks
- block_size = min((pyramid_width - padding) // max_companies - padding, (img_height - padding) // total_groups - padding)
-
- # Calculate the size of each block based on the width and height of the image and the number of blocks
- block_width = (pyramid_width - padding) // max_companies - padding
- block_height = (img_height - padding) // total_groups - padding
-
- # Calculate the total width and height of the blocks (including padding)
- total_width = max_companies * (block_width + padding)
- total_height = total_groups * (block_height + padding)
-
- # Calculate the starting position for the first block
- start_x = (pyramid_width - total_width) // 2
- start_y = (img_height - total_height) // 2
-
-
- # Create an image big enough to hold the pyramid
- img = Image.new('RGB', (pyramid_width, img_height), "white")
- d = ImageDraw.Draw(img)
-
- def wrap_text(text, max_length):
- words = text.split()
- lines = []
- current_line = []
-
- for word in words:
- if len(' '.join(current_line + [word])) <= max_length:
- current_line.append(word)
- else:
- lines.append(' '.join(current_line))
- current_line = [word]
- lines.append(' '.join(current_line))
-
- return '\n'.join(lines)
-
- min_font_size = 100_000_000
-
- for i, row in enumerate(companies):
- for j, company in enumerate(row):
- font_size = min(block_width // (len(company) // 2 + 1), block_height // 2)
- if font_size < min_font_size:
- min_font_size = font_size
-
- num_dividends = 0
+ document = Document("assets/doc_template.docx")
- # Loop over each level of the pyramid
- for i, row in enumerate(companies):
- for j, company in enumerate(row):
- # Calculate the position of the block
- x = start_x + j * (block_width + padding) + (max_companies - len(row)) * (block_width + padding) // 2
- y = start_y + i * (block_size + padding)
-
- # Calculate the color of the block
- if company[-1] == "*":
- block_color = dividend_color
- num_dividends += 1
+ advisor = self.advisor_string.get()
+ today = datetime.today()
+
+ document.paragraphs[0].text = document.paragraphs[0].text.replace("{{ADVISOR}}", advisor)
+
+ to_replace = document.paragraphs[3].text
+
+ to_replace = to_replace.replace("{{MONTH}}" , today.strftime("%B"))
+ to_replace = to_replace.replace("{{YEAR}}" , today.strftime("%Y"))
+
+ document.paragraphs[3].text = to_replace
+
+ for company, ticker, dividend, currency, desc in zip(selected_df["Company Name"], selected_df["Symbol"], selected_df["Dividend?"], selected_df["Currency"], selected_df["Blurb"]):
+ extra_char = ""
+ if dividend.strip() == "Y":
+ extra_char = "***"
+
+ new_paragraph = document.add_paragraph()
+ temp_run = new_paragraph.add_run(f'{company}{extra_char} ({ticker})')
+ temp_run.bold = True
+ new_paragraph.alignment = WD_ALIGN_PARAGRAPH.LEFT
+ document.paragraphs.pop()
+ desc_paragraph = document.add_paragraph(f"\n{desc}\n")
+ desc_paragraph.alignment = WD_ALIGN_PARAGRAPH.JUSTIFY
+ desc_paragraph.bold = False
+
+ document.save(word_file_path)
+
+ if self.generate_pyramid_var.get() == 0:
+ None
+ else:
+ image_file_path = asksaveasfilename(defaultextension=".png", filetypes=[("Image files","*.png")])
+
+ company_scores = {}
+
+ # For each 'Company Name', company_scores[company] = 'Weighting' for that company
+ for company, ticker, weighting, dividend, currency in zip(selected_df["Company Name"], selected_df["Symbol"], selected_df["Weighting"], selected_df["Dividend?"], selected_df["Currency"]):
+ extra_char = ""
+ if dividend.strip() == "Y":
+ extra_char = "*"
+ if currency == "CAD":
+ canadian_companies.append(f'{company} ({ticker}){extra_char}')
+ company_scores[f'{company} ({ticker}){extra_char}'] = weighting
+
+ from PIL import Image, ImageDraw, ImageFont
+ import math
+ # Sort the dictionary by score in ascending order
+ company_scores = {k: v for k, v in sorted(company_scores.items(), key=lambda item: item[1])}
+
+ # Group the companies by their scores
+ pyramid = {i: [] for i in range(1, 7)}
+ for company, score in company_scores.items():
+ pyramid[score].append(company)
+
+ # Initialize some parameters
+ img_width = 3300
+ pyramid_width = 3000
+ img_height = 1550 #2550
+ overall_height = 2550
+ block_color = (3, 37, 126) # Blue color
+ dividend_color = block_color #(1,50,32) # Green color
+ font_color = (255, 255, 255) # White color
+ normal_block = (3, 37, 126) # Dark Blue
+ padding = 15 # Padding around blocks
+ radius = 10
+ logo_padding = 100 # top padding for logo
+ pyramid_padding_bottom = 350 # bottom padding for pyramid
+ max_companies_in_row = 10
+ base_path = "assets"
+ primary_font = path.join(base_path, "baskerville.ttf")
+ secondary_font = path.join(base_path, "gill_sans_bold.ttf")
+
+ # Group the companies by their scores
+ pyramid = {}
+ for company, score in company_scores.items():
+ if score not in pyramid:
+ pyramid[score] = []
+ pyramid[score].append(company)
+
+ # Sort the pyramid keys in ascending order to have a pyramid shape
+ sorted_keys = sorted(pyramid.keys())
+
+ companies = []
+
+
+ for key in sorted_keys:
+ if pyramid[key] == []:
+ continue
+ if len(pyramid[key])<=max_companies_in_row:
+ companies.append(pyramid[key])
else:
- block_color = normal_block
+ temp_list = []
+ for x in range(len(pyramid[key])//max_companies_in_row):
+ temp_list.append(pyramid[key][x*max_companies_in_row:max_companies_in_row])
+
+ temp_list.append(pyramid[key][::-1][:len(pyramid[key])%max_companies_in_row])
+ for row in temp_list[::-1]:
+ if row != []:
+ companies.append(row)
+
+ everything_list = []
+ for row in companies:
+ if row == []:
+ continue
+ for company in row:
+ everything_list.append(company)
+
+ companies = everything_list
+ print(f"Still have {len(companies)}")
- # Draw the block
- d.rounded_rectangle([x, y, x + block_width, y + block_height], fill=block_color, radius=radius)
+ companies = pyramid_list(companies)
- # Adjust font size based on the length of the company name and block size
- font_size = min_font_size #min(block_width // (len(company) // 2 + 1), block_height // 2)
- fnt = ImageFont.truetype(secondary_font, font_size)
+ for company_row in companies:
+ print(company_row)
- # Implement word wrap for the company name
- wrapped_company = wrap_text(company, (block_width // font_size)*1.5)
-
- # Draw the company name
- bbox = d.textbbox((x, y), wrapped_company, font=fnt)
- text_width = bbox[2] - bbox[0]
- text_height = bbox[3] - bbox[1]
- text_x = x + (block_width - text_width) // 2
- text_y = y + (block_height - text_height) // 2
- d.text((text_x, text_y), wrapped_company, font=fnt, fill=font_color, align='center', spacing=10)
-
-
- print(image_file_path)
-
- # Save the image
- if not image_file_path:
- return
+ # Calculate the maximum number of companies in a group (this will be the width of your pyramid)
+ max_companies = max(len(row) for row in companies) #max(len(v) for v in pyramid.values())
- image_logo = Image.open("./assets/GentryCapitalRGB.jpg")
- _, l_height = image_logo.size
+ # Calculate the total number of groups (the height of your pyramid)
+ total_groups = len(companies) #len(pyramid)
- new_img = Image.new("RGB", (img_width, l_height), "white")
- left = (new_img.width - image_logo.width) // 2
- top = (new_img.height - image_logo.height) // 2
+ # Calculate the size of each block based on the width and height of the image and the number of blocks
+ block_size = min((pyramid_width - padding) // max_companies - padding, (img_height - padding) // total_groups - padding)
- new_img.paste(image_logo, (left, top))
+ # Calculate the size of each block based on the width and height of the image and the number of blocks
+ block_width = (pyramid_width - padding) // max_companies - padding
+ block_height = (img_height - padding) // total_groups - padding
+
+ # Calculate the total width and height of the blocks (including padding)
+ total_width = max_companies * (block_width + padding)
+ total_height = total_groups * (block_height + padding)
+
+ # Calculate the starting position for the first block
+ start_x = (pyramid_width - total_width) // 2
+ start_y = (img_height - total_height) // 2
+
+
+ # Create an image big enough to hold the pyramid
+ img = Image.new('RGB', (pyramid_width, img_height), "white")
+ d = ImageDraw.Draw(img)
+
+ def wrap_text(text, max_length):
+ words = text.split()
+ lines = []
+ current_line = []
+
+ for word in words:
+ if len(' '.join(current_line + [word])) <= max_length:
+ current_line.append(word)
+ else:
+ lines.append(' '.join(current_line))
+ current_line = [word]
+ lines.append(' '.join(current_line))
+
+ return '\n'.join(lines)
+
+ min_font_size = 100_000_000
+
+ for i, row in enumerate(companies):
+ for j, company in enumerate(row):
+ font_size = min(block_width // (len(company) // 2 + 1), block_height // 2)
+ if font_size < min_font_size:
+ min_font_size = font_size
+
+ num_dividends = 0
+
+ # Loop over each level of the pyramid
+ for i, row in enumerate(companies):
+ for j, company in enumerate(row):
+ # Calculate the position of the block
+ x = start_x + j * (block_width + padding) + (max_companies - len(row)) * (block_width + padding) // 2
+ y = start_y + i * (block_size + padding)
+
+ # Calculate the color of the block
+ if company[-1] == "*":
+ block_color = dividend_color
+ num_dividends += 1
+ else:
+ block_color = normal_block
+
+ # Draw the block
+ d.rounded_rectangle([x, y, x + block_width, y + block_height], fill=block_color, radius=radius)
+
+ # Adjust font size based on the length of the company name and block size
+ font_size = min_font_size #min(block_width // (len(company) // 2 + 1), block_height // 2)
+ fnt = ImageFont.truetype(secondary_font, font_size)
+
+ # Implement word wrap for the company name
+ wrapped_company = wrap_text(company, (block_width // font_size)*1.5)
+
+ # Draw the company name
+ bbox = d.textbbox((x, y), wrapped_company, font=fnt)
+ text_width = bbox[2] - bbox[0]
+ text_height = bbox[3] - bbox[1]
+ text_x = x + (block_width - text_width) // 2
+ text_y = y + (block_height - text_height) // 2
+ d.text((text_x, text_y), wrapped_company, font=fnt, fill=font_color, align='center', spacing=10)
+
+
+ print(image_file_path)
+
+ # Save the image
+ if not image_file_path:
+ return
- final_img = Image.new("RGB", (img_width, overall_height), "white")
- final_img.paste(new_img, (0, logo_padding))
-
- text1 = self.pyramid_title_string.get()
- text2 = f"({num_dividends} Dividend payors - all identified by asterisk)"
- text3 = "FOR INTERNAL USE ONLY"
- text4 = datetime.today().strftime("%B %d, %Y")
- text5 = ""
-
- if self.prepared_for_string.get() != "":
- print(f"Prepared for string present - {self.prepared_for_string.get()}")
- text5 += f"Prepared for {self.prepared_for_string.get()}"
- if self.advisor_string.get() != "":
- advisor = self.advisor_string.get()
- if text5 != "":
- tmp_var = [text5, f"By {advisor}"]
- text5 = tmp_var
- else:
- text5 = f"By {advisor}"
+ image_logo = Image.open("./assets/GentryCapitalRGB.jpg")
+ _, l_height = image_logo.size
+
+ new_img = Image.new("RGB", (img_width, l_height), "white")
+ left = (new_img.width - image_logo.width) // 2
+ top = (new_img.height - image_logo.height) // 2
+
+ new_img.paste(image_logo, (left, top))
+
+ final_img = Image.new("RGB", (img_width, overall_height), "white")
+ final_img.paste(new_img, (0, logo_padding))
+
+ text1 = self.pyramid_title_string.get()
+ text2 = f"({num_dividends} Dividend payors - all identified by asterisk)"
+ text3 = "FOR INTERNAL USE ONLY"
+ text4 = datetime.today().strftime("%B %d, %Y")
+ text5 = ""
+
+ if self.prepared_for_string.get() != "":
+ print(f"Prepared for string present - {self.prepared_for_string.get()}")
+ text5 += f"Prepared for {self.prepared_for_string.get()}"
+ if self.advisor_string.get() != "":
+ advisor = self.advisor_string.get()
+ if text5 != "":
+ tmp_var = [text5, f"By {advisor}"]
+ text5 = tmp_var
+ else:
+ text5 = f"By {advisor}"
- print(text5)
- draw = ImageDraw.Draw(final_img)
- font_size = 60
+ print(text5)
+ draw = ImageDraw.Draw(final_img)
+ font_size = 60
- font1 = ImageFont.truetype(primary_font, font_size)
- text_width1 = draw.textlength(text1, font=font1)
- while text_width1 > final_img.width:
- font_size -= 1
font1 = ImageFont.truetype(primary_font, font_size)
text_width1 = draw.textlength(text1, font=font1)
+ while text_width1 > final_img.width:
+ font_size -= 1
+ font1 = ImageFont.truetype(primary_font, font_size)
+ text_width1 = draw.textlength(text1, font=font1)
- font2_size = font_size - 10
- font2 = ImageFont.truetype(primary_font, font2_size)
- text_width2 = draw.textlength(text2, font=font2)
- while text_width2 > final_img.width:
- font_size -= 1
+ font2_size = font_size - 10
font2 = ImageFont.truetype(primary_font, font2_size)
text_width2 = draw.textlength(text2, font=font2)
+ while text_width2 > final_img.width:
+ font_size -= 1
+ font2 = ImageFont.truetype(primary_font, font2_size)
+ text_width2 = draw.textlength(text2, font=font2)
- new_font_size = 30
- font3 = ImageFont.truetype(secondary_font, new_font_size)
- _,t,_,b = draw.textbbox((100,100), text3, font=font3)
- while (b-t) < (0.4*pyramid_padding_bottom):
- new_font_size += 1
+ new_font_size = 30
font3 = ImageFont.truetype(secondary_font, new_font_size)
_,t,_,b = draw.textbbox((100,100), text3, font=font3)
- text_width3 = draw.textlength(text3, font=font3)
- while text_width3 > (final_img.width*0.8):
- new_font_size -= 1
- font3 = ImageFont.truetype(secondary_font, new_font_size)
+ while (b-t) < (0.4*pyramid_padding_bottom):
+ new_font_size += 1
+ font3 = ImageFont.truetype(secondary_font, new_font_size)
+ _,t,_,b = draw.textbbox((100,100), text3, font=font3)
text_width3 = draw.textlength(text3, font=font3)
- _,t,_,b = draw.textbbox((100,100),text3,font=font3)
+ while text_width3 > (final_img.width*0.8):
+ new_font_size -= 1
+ font3 = ImageFont.truetype(secondary_font, new_font_size)
+ text_width3 = draw.textlength(text3, font=font3)
+ _,t,_,b = draw.textbbox((100,100),text3,font=font3)
- font4_size = font2_size - 5
- font4 = ImageFont.truetype(primary_font, font4_size)
- text_width4 = draw.textlength(text4, font=font4)
+ font4_size = font2_size - 5
+ font4 = ImageFont.truetype(primary_font, font4_size)
+ text_width4 = draw.textlength(text4, font=font4)
- start_x1 = (final_img.width - text_width1) // 2
- start_y1 = image_logo.height + logo_padding + 50
+ start_x1 = (final_img.width - text_width1) // 2
+ start_y1 = image_logo.height + logo_padding + 50
- start_x2 = (final_img.width - text_width2) // 2
- start_y2 = image_logo.height + logo_padding + 125
+ start_x2 = (final_img.width - text_width2) // 2
+ start_y2 = image_logo.height + logo_padding + 125
- start_x3 = (final_img.width - text_width3) // 2
- start_y3 = (final_img.height - pyramid_padding_bottom + ((pyramid_padding_bottom)-(b-t))//2)
+ start_x3 = (final_img.width - text_width3) // 2
+ start_y3 = (final_img.height - pyramid_padding_bottom + ((pyramid_padding_bottom)-(b-t))//2)
- start_x4 = (final_img.width - text_width4) - 150
- start_y4 = logo_padding + 30
+ start_x4 = (final_img.width - text_width4) - 150
+ start_y4 = logo_padding + 30
- start_x5 = 150
- start_y5 = start_y4
-
- draw.text((start_x1, start_y1), text1, font=font1, fill="black")
- draw.text((start_x2, start_y2), text2, font=font2, fill="black")
- draw.text((start_x3, start_y3), text3, font=font3, fill=(158,161,162))
- draw.text((start_x4, start_y4), text4, font=font4, fill="black")
- if type(text5) == str:
- draw.text((start_x5, start_y5), text5, font=font4, fill="black")
- else:
- for idx, text2write in enumerate(text5):
- draw.text((start_x5, start_y5 + (idx*69)), text2write, font=font4, fill="black")
-
- final_img.paste(img, ((img_width-pyramid_width)//2, l_height + (overall_height - l_height - img_height) - pyramid_padding_bottom) )
-
- # draw = ImageDraw.Draw(final_img)
- # watermark_text = "FOR INTERNAL USE ONLY"
- # watermark_font = ImageFont.truetype(secondary_font, 72)
- # watermark_width = draw.textlength(watermark_text, watermark_font)
- # draw.text(
- # ((final_img.width - watermark_width)//2,overall_height-pyramid_padding_bottom),
- # watermark_text,
- # font=watermark_font,
- # align='center'
- # )
-
- final_img.save(image_file_path)
- final_img.show()
- #img.show()
-
- """if not pyramid_file_path:
- return
-
- with pd.ExcelWriter(pyramid_file_path) as writer:
- selected_df.to_excel(writer, index=False)"""
-
- #self.master.quit()
+ start_x5 = 150
+ start_y5 = start_y4
+
+ draw.text((start_x1, start_y1), text1, font=font1, fill="black")
+ draw.text((start_x2, start_y2), text2, font=font2, fill="black")
+ draw.text((start_x3, start_y3), text3, font=font3, fill=(158,161,162))
+ draw.text((start_x4, start_y4), text4, font=font4, fill="black")
+ if type(text5) == str:
+ draw.text((start_x5, start_y5), text5, font=font4, fill="black")
+ else:
+ for idx, text2write in enumerate(text5):
+ draw.text((start_x5, start_y5 + (idx*69)), text2write, font=font4, fill="black")
+
+ final_img.paste(img, ((img_width-pyramid_width)//2, l_height + (overall_height - l_height - img_height) - pyramid_padding_bottom) )
+
+ # draw = ImageDraw.Draw(final_img)
+ # watermark_text = "FOR INTERNAL USE ONLY"
+ # watermark_font = ImageFont.truetype(secondary_font, 72)
+ # watermark_width = draw.textlength(watermark_text, watermark_font)
+ # draw.text(
+ # ((final_img.width - watermark_width)//2,overall_height-pyramid_padding_bottom),
+ # watermark_text,
+ # font=watermark_font,
+ # align='center'
+ # )
+
+ final_img.save(image_file_path)
+ final_img.show()
+ #img.show()
+
+ """if not pyramid_file_path:
+ return
+
+ with pd.ExcelWriter(pyramid_file_path) as writer:
+ selected_df.to_excel(writer, index=False)"""
+
+ #self.master.quit()
if __name__ == "__main__":
import sentry_sdk