factories.py 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300
  1. """Calculate production in different factories"""
  2. from rival_regions_calc import Item, WorkProduction
  3. DEPARTMENTS = {
  4. 'oil': 1.5,
  5. 'gold': 0.4,
  6. 'diamond': 0.1,
  7. 'uranium': 0,
  8. 'ore': 0,
  9. }
  10. PRICE = {
  11. 'oil': 255,
  12. 'gold': 0,
  13. 'diamond': 1650000,
  14. 'uranium': 2300,
  15. 'ore': 235,
  16. }
  17. REGIONS = {
  18. 4001: {
  19. 'name': 'Northern Netherlands',
  20. 'resources': {
  21. 'gold': 379,
  22. 'oil': 223,
  23. 'ore': 247,
  24. # 'uranium': 2,
  25. 'uranium': 25,
  26. 'diamond': 5,
  27. },
  28. 'tax': {
  29. 'gold': 15,
  30. 'oil': 15,
  31. 'ore': 15,
  32. 'uranium': 15,
  33. 'diamond': 15,
  34. },
  35. },
  36. 4002: {
  37. 'name': 'Eastern Netherlands',
  38. 'resources': {
  39. 'gold': 359,
  40. # 'oil': 266,
  41. 'oil': 371,
  42. 'ore': 250,
  43. 'uranium': 2,
  44. 'diamond': 2,
  45. },
  46. 'tax': {
  47. 'gold': 15,
  48. 'oil': 2,
  49. 'ore': 15,
  50. 'uranium': 15,
  51. 'diamond': 15,
  52. },
  53. },
  54. 4003: {
  55. 'name': 'Western Netherlands',
  56. 'resources': {
  57. 'gold': 372,
  58. 'oil': 296,
  59. 'ore': 230,
  60. 'uranium': 2,
  61. 'diamond': 2,
  62. },
  63. 'tax': {
  64. 'gold': 15,
  65. 'oil': 15,
  66. 'ore': 15,
  67. 'uranium': 15,
  68. 'diamond': 15,
  69. },
  70. },
  71. 4004: {
  72. 'name': 'Southern Netherlands',
  73. 'resources': {
  74. 'gold': 366,
  75. 'oil': 296,
  76. 'ore': 211,
  77. 'uranium': 2,
  78. 'diamond': 2,
  79. },
  80. 'tax': {
  81. 'gold': 15,
  82. 'oil': 15,
  83. 'ore': 15,
  84. 'uranium': 15,
  85. 'diamond': 15,
  86. },
  87. },
  88. 4008: {
  89. 'name': 'Amsterdam',
  90. 'resources': {
  91. 'gold': 418,
  92. 'oil': 307,
  93. 'ore': 303,
  94. 'uranium': 4,
  95. 'diamond': 13,
  96. },
  97. 'tax': {
  98. 'gold': 15,
  99. 'oil': 15,
  100. 'ore': 15,
  101. 'uranium': 15,
  102. 'diamond': 15,
  103. },
  104. },
  105. 4801: {
  106. 'name': 'Luxembourg',
  107. 'resources': {
  108. 'gold': 435,
  109. 'oil': 283,
  110. 'ore': 267,
  111. 'uranium': 2,
  112. 'diamond': 2,
  113. },
  114. 'tax': {
  115. 'gold': 15,
  116. 'oil': 15,
  117. 'ore': 15,
  118. 'uranium': 15,
  119. 'diamond': 15,
  120. },
  121. }
  122. }
  123. RESOURCES = {
  124. 'oil': {
  125. 45760: {
  126. 'name': 'OIL RAFINERY',
  127. 'level': 185,
  128. 'region_id': 4002,
  129. 'wage': 100,
  130. },
  131. 32282: {
  132. 'name': 'OIL - Lorenzo Industries',
  133. 'level': 157,
  134. 'region_id': 4801,
  135. 'wage': 100,
  136. },
  137. 59465: {
  138. 'name': 'Verenigde Nederlanden Oil',
  139. 'level': 144,
  140. 'region_id': 4003,
  141. 'wage': 99,
  142. },
  143. 90420: {
  144. 'name': 'Shelby oil',
  145. 'level': 124,
  146. 'region_id': 4002,
  147. 'wage': 95,
  148. },
  149. 61361: {
  150. 'name': 'Koninklijke Olie',
  151. 'level': 120,
  152. 'region_id': 4001,
  153. 'wage': 100,
  154. 'fixed_wage': 150500500,
  155. },
  156. 60828: {
  157. 'name': 'Austins wegenolie',
  158. 'level': 117,
  159. 'region_id': 4008,
  160. 'wage': 100,
  161. 'fixed_wage': 130000000,
  162. },
  163. },
  164. 'ore': {
  165. 32145: {
  166. 'name': 'DS ore',
  167. 'level': 139,
  168. 'region_id': 4801,
  169. 'wage': 99,
  170. },
  171. 48500: {
  172. 'name': 'Shelby ore',
  173. 'level': 138,
  174. 'region_id': 4004,
  175. 'wage': 95,
  176. },
  177. 29239: {
  178. 'name': 'Nederlands Ore EN',
  179. 'level': 73,
  180. 'region_id': 4002,
  181. 'wage': 80,
  182. },
  183. },
  184. 'uranium': {
  185. 34997: {
  186. 'name': 'Quite a serious power plant!',
  187. 'level': 147,
  188. 'region_id': 4001,
  189. 'wage': 100,
  190. 'fixed_wage': 350000000,
  191. },
  192. },
  193. 'diamond': {
  194. 33263: {
  195. 'name': 'Shelby Diamonds #2',
  196. 'level': 148,
  197. 'region_id': 4008,
  198. 'wage': 95,
  199. },
  200. 63317: {
  201. 'name': 'Vone I',
  202. 'level': 30,
  203. 'region_id': 4008,
  204. 'wage': 80,
  205. },
  206. },
  207. }
  208. WP = WorkProduction()
  209. WP.user_level = 95
  210. WP.work_exp = 80000 + 200 * 350
  211. WP.nation_bonus = True
  212. WP.profit_share = 75
  213. def calculate_wage(resource_name, factory):
  214. """Calculate production"""
  215. if 'fixed_wage' in factory:
  216. return factory['fixed_wage']
  217. resource = Item(resource_name)
  218. WP.resource = resource
  219. WP.factory_level = factory['level']
  220. WP.department_bonus = DEPARTMENTS[resource_name]
  221. WP.resource_max = REGIONS[factory['region_id']]['resources'][resource_name]
  222. WP.wage_percentage = factory['wage']
  223. WP.tax_rate = REGIONS[factory['region_id']]['tax'][resource_name]
  224. WP.calculate()
  225. return WP.wage() * PRICE[resource_name]
  226. def all_factories():
  227. """return all factories"""
  228. factory_dict = {}
  229. for factories in RESOURCES.values():
  230. factory_dict.update(factories)
  231. return factory_dict
  232. def print_per_resource():
  233. """Print wages per resource"""
  234. for resource_name, factories in RESOURCES.items():
  235. print(resource_name)
  236. for factory in sorted(factories.values(), key=lambda k: k['wage'], reverse=True):
  237. print('{:30} {:>12}'.format(
  238. factory['name'],
  239. int(factory['wage']),
  240. ))
  241. def print_all():
  242. """Print wages per resource"""
  243. factory_dict = all_factories()
  244. print("total")
  245. sorted_factories = sorted(factory_dict.values(), key=lambda k: k['wage'], reverse=True)
  246. top_wage = sorted_factories[0]['wage']
  247. for factory in sorted_factories:
  248. print('{:28} {:>11,} {:>3}'.format(
  249. factory['name'],
  250. int(factory['wage']),
  251. int(100 / top_wage * factory['wage']),
  252. ).replace(',', '.'))
  253. def print_factory_list():
  254. """print all factories"""
  255. for resource_name, factories in RESOURCES.items():
  256. print('[rr]{}[/rr]'.format(resource_name.capitalize()))
  257. for factory_id, factory in factories.items():
  258. print('[url=https://rivalregions.com/#factory/index/{}]Desktop[/url] [url=https://m.rivalregions.com/#factory/index/{}]Mobile[/url] {}, {} '.format(
  259. factory_id,
  260. factory_id,
  261. factory['name'],
  262. REGIONS[factory['region_id']]['name'],
  263. ))
  264. def main():
  265. """Main function"""
  266. for resource_name, factories in RESOURCES.items():
  267. for factory_id, factory in factories.items():
  268. wage = calculate_wage(resource_name, factory)
  269. RESOURCES[resource_name][factory_id]['wage'] = wage
  270. # print_per_resource()
  271. print_all()
  272. # print_factory_list()
  273. if __name__ == '__main__':
  274. main()