Allow to find files in pyinstaller

1 files changed, 6 insertions, 1 deletions

diff --git a/helpers/__init__.py b/helpers/__init__.py
index c42b378..7dbffdd 100644
--- a/helpers/__init__.py
+++ b/helpers/__init__.py
@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 from pygame import *
+import sys
 
 class Action:
     action_types = [
@@ -106,7 +107,11 @@ class Key:
     def draw(self, background_surface):
         self.surface = self.square().surface()
 
-        police = font.Font("Ubuntu-Regular.ttf", 14)
+        if getattr(sys, 'frozen', False):
+            police = font.Font(sys._MEIPASS + "/Ubuntu-Regular.ttf", 14)
+        else:
+            police = font.Font("Ubuntu-Regular.ttf", 14)
+
         text = police.render(self.key_sym, True, (0,0,0))
         self.surface.blit(text, (5,5))
         background_surface.blit(self.surface, self.position)