Added generate script root directory resolve to launch it from language directories

1 files changed, 3 insertions, 3 deletions

diff --git a/generate b/generate
index fef0aed..5773fd8 100755
--- a/generate
+++ b/generate
@@ -10,12 +10,12 @@ import requests
 from bs4 import BeautifulSoup
 
 
-LANGUAGES_FILENAME = 'languages.json'
+ROOT_DIR           = Path(__file__).resolve().parent
+LANGUAGES_FILENAME = ROOT_DIR / 'languages.json'
 URL_FORMAT         = 'http://projecteuler.net/problem={index}'
 LINE_WRAP          = 89
 PROBLEM_PADDING    = 3
 
-
 class Problem:
     def __init__(self, index: int, language: dict):
         self.index = index
@@ -35,7 +35,7 @@ class Problem:
 
     def write(self):
         file_name = f'{self.index:03}-{self._slug}.{self.language["extension"]}'
-        file_path = Path(self.language['name']) / file_name
+        file_path = ROOT_DIR / self.language['name'] / file_name
         if file_path.exists():
             raise FileExistsError(f'{file_path} already exists')
         file_path.parent.mkdir(exist_ok=True)