reworked/calc.py

@@ -1,14 +1,12 @@
-global expanded
+global expanded ; expanded = {}
 def expand_recipe(recipe, recipes, main = False):
 	global expanded
-	if main:
-		expanded = {}
 
 	ok = False
 	part = 1 # Одна часть
 	while not ok:
 		ok = True
-		vol_in = 0 # Объём мин. рецепта (вход)
+		min_vol = 0 # Объём мин. рецепта (вход)
 		# Перебираем элементы
 		for el in recipe:
 			# Если составное
@@ -20,31 +18,35 @@ def expand_recipe(recipe, recipes, main = False):
 					expanded = {}
 					break
 				else:
-					vol_in += expand_recipe(recipes[el].comps, recipes)
+					min_vol += expand_recipe(recipes[el].comps, recipes)
 			else:
 				if el in expanded:
 					expanded[el] += recipe[el]*part
 				else:
 					expanded[el] = recipe[el]*part
 
-				vol_in += recipe[el]*part
+				min_vol += recipe[el]*part
 
 	if main:
-		return expanded, vol_in, part
+		return expanded, min_vol, part
 	else:
-		return vol_in
+		return min_vol
 
 def calc(element, amount, recipes):
 	# Получаем характеристику элемента
-	recipe, vol_out = recipes[element].comps, recipes[element].out
+	recipe, out = recipes[element].comps, recipes[element].out
 	# Расчитываем минимальный рецепт
-	expanded, vol_in, part = expand_recipe(recipe, recipes, True)
+	expanded, vol, part = expand_recipe(recipe, recipes, True)
 
-	need = amount//vol_in
+	# Домнажаем на сколько нужно
+	need = amount//vol
 	for i in expanded:
 		expanded[i] = expanded[i]*need
-	vol_in *= need
-	vol_out *= part*need
+	out = part*out*need
 
-	return expanded, vol_in, vol_out
+	return expanded, out
 
+
+from parse import *
+#print( load_recipes() )
+print( calc('Leporazine', 100, load_recipes()) )

reworked/db.py

@@ -1,43 +1,20 @@
 import os
 import json
-from reag__ import reag__
 
 if not os.path.exists('db.json'):
 	db = {}
 	js = json.dumps(db, indent=2)
-	with open('db.json', 'w') as outfile:
+	with open("db.json", "w") as outfile:
 		outfile.write(js)
 	print('Created new db.json')
-if not os.path.exists('raw_db.json'):
-  db = {}
-  js = json.dumps(db, indent=2)
-  with open('raw_db.json', 'w') as outfile:
-    outfile.write(js)
-  print('Created new raw_db.json')
 
 
-def read_db(file):
-	with open(file, 'r', encoding='utf-8') as openfile:
+def read_db(file = 'db.json'):
+	with open(file, "r", encoding="utf-8") as openfile:
 		db = json.load(openfile)
 	return db
 
-def write_db(db, file):
+def write_db(db, file = 'db.json'):
 	js = json.dumps(db, indent=2, ensure_ascii=False)
-	with open(file, 'w', encoding='utf-8') as outfile:
+	with open(file, "w", encoding="utf-8") as outfile:
 		outfile.write(js)
-
-
-
-def save(db, file):
-	raw = {}
-	for el in db:
-		class_data = db[el].get_all()
-		raw[el] = class_data
-	write_db(raw, file)
-
-def load(file):
-	raw = read_db(file)
-	db = {}
-	for el in raw:
-		db[el] = reag__(raw[el][0], raw[el][1], raw[el][2])
-	return db

reworked/parse.py

@@ -2,10 +2,11 @@ from requests import get
 from yaml import load, SafeLoader
 from reag__ import reag__
 
-#### Локализация ####
+def parse_yml(url = 'https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/Resources/Prototypes/Recipes/Reactions/medicine.yml'):
+	yml = load(get(url).content.decode('utf-8'), Loader=SafeLoader)
+	return yml
 
-def parse_ftl(el, prefix = 'https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/Resources/Locale/ru-RU/reagents/meta'):
-	url = f'{prefix}/{el}.ftl'
+def parse_ftl(url = 'https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/Resources/Locale/ru-RU/reagents/meta/medicine.ftl'):
 	raw = get(url).content.decode('utf-8')
 	locales = {}
 	for i in raw.splitlines():
@@ -16,36 +17,20 @@ def parse_ftl(el, prefix = 'https://raw.githubusercontent.com/SerbiaStrong-220/s
 			locales[name] = locale
 	return locales
 
-def load_locales(locales_url):
-	locales = {}
-	for el in locales_url:
-		locales = locales | parse_ftl(el)
-	return locales
-
-#### Рецепты ####
-
-def parse_yml(el, prefix = 'https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/Resources/Prototypes/Recipes/Reactions'):
-	url = f'{prefix}/{el}.yml'
-	yml = load(get(url).content.decode('utf-8'), Loader=SafeLoader)
-	return yml
-
-def load_recipes(recipes_url, prefix = 'https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/Resources/Prototypes/Recipes/Reactions'):
-	for el in recipes_url:
-		yml = parse_yml(el, prefix)
-		recipes = {}
-		for element in yml:
-			product = element["id"]
-			comps = {}
-			for elem in element["reactants"]:
-				comps[elem] = element["reactants"][elem]["amount"]
-			for id, value in element["products"].items():
-				out =  value
-				recipes[product] = reag__(comps=comps, out=out, category=el)
+def load_recipes(url = 'https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/Resources/Prototypes/Recipes/Reactions/medicine.yml', category = '-'):
+	yml = parse_yml(url)
+	recipes = {}
+	for element in yml:
+		product = element["id"]
+		comps = {}
+		for elem in element["reactants"]:
+			comps[elem] = element["reactants"][elem]["amount"]
+		for id, value in element["products"].items():
+			out =  value
+			recipes[product] = reag__(category=category, comps=comps, out=out)
 	return recipes
 
 
-#### Локализируем ####
-
 def localize(recipes, locale):
 	loc_recipes = {}
 	# Итерируем элементы
@@ -72,3 +57,17 @@ def localize(recipes, locale):
 			loc_recipes[element] = recipes[element]
 
 	return loc_recipes
+
+
+
+
+recipes = load_recipes()
+locales = parse_ftl()
+
+recipes = localize(recipes, locales)
+
+from icecream import ic
+ic.configureOutput(prefix='')
+for i in recipes:
+	el = [i, recipes[i].comps]
+	ic(el)

reworked/reag__.py

@@ -1,11 +1,11 @@
 class reag__:
-	def __init__(self, comps, out, category = '-'):
+	def __init__(self, category: str = '-', comps = {}, out: int = 0):
+		# medicine
+		self.category = category
 		# {'инапровалин': 1, 'углерод': 1}
 		self.comps = comps
 		# 2
 		self.out = out
-		# medicine
-		self.category = category
 	def get_all(self):
-		return [self.comps, self.out, self.category]
+		return [self.category, self.comps, self.out]
 

reworked/update_db.py

@@ -1,23 +0,0 @@
-from parse import *
-from calc import calc
-from db import *
-
-'''
-# Загружаем локализацию
-locales_url = ['medicine', 'chemicals']
-locales = load_locales(locales_url)
-
-# Загружаем сырые рецепты
-recipes_url = ['medicine']
-raw_recipes = load_recipes(recipes_url)
-
-# Локализируем
-recipes = localize(raw_recipes, locales)
-# Сохранаяем данные
-save(recipes, 'raw_db.json')
-'''
-
-rec = load('raw_db.json')
-
-# Делаем предрасчёты
-#calculated =