import numpy as np
import matplotlib.pyplot as plt
import sys

lang = "en"
#lang = "tr"

Turkmenistan      = np.array([1.3   ,0     ,0     ,0     ,0     ,0     ,0     ,0     ])
Russia_Bluestream = np.array([16    ,0     ,0     ,0     ,0     ,0     ,0     ,0     ])
Russia_Turkstream = np.array([5.75  ,0     ,0     ,0     ,0     ,0     ,0     ,0     ])
Iran              = np.array([9.6   ,0     ,0     ,0     ,0     ,0     ,0     ,0     ])
Azerbaijan        = np.array([6     ,6     ,6     ,6     ,6     ,6     ,6     ,6     ])
year              =          ['2026','2027','2028','2029','2030','2031','2032','2033']

if lang == "en":
 plt.title ("Contracts for pipeline gas supply to Turkey")
 plt.ylabel("Billion cubic metres")
 label_AZ       = "Azerbaijan"
 label_IR       = "Iran"
 label_RU_TS    = "Russia Turkstream"
 label_RU_BS    = "Russia Bluestream"
 label_TM       = "Turkmenistan"
 Source         = "Cite"
elif lang == "tr":
 plt.title ("Türkiye'ye boru hattı gazı tedarik sözleşmeleri")
 plt.ylabel("Milyar metre küp")
 label_AZ       = "Azarbeycan"
 label_IR       = "İran"
 label_RU_TS    = "Rusya TürkAkım"
 label_RU_BS    = "Rusya MaviAkım"
 label_TM       = "Türkmenistan"
 Source         = ""
else:
 print("Unknown language " + lang)
 sys.exit()
 
plt.xlabel(Source + " reuters.com/business/energy/turkey-extended-russia-gas-contracts-by-year-eying-us-gas-investment-2025-12-04\nreuters.com/business/energy/turkey-seals-11-bcm-natural-gas-accord-with-azerbaijan-minister-2021-10-15\nreuters.com/business/energy/turkey-signs-10-yr-gas-deal-with-oman-turkish-energy-minister-2023-01-30", fontsize = 'x-small', color = 'grey')

#Don't need the year dash marks
plt.tick_params(bottom=False,left=False)

#Put last to finish at bottom
plt.bar(year, Azerbaijan,                                                                            width=1, color="yellow")
plt.bar(year, Iran,                      bottom=Azerbaijan,                                          width=1, color="green") 
plt.bar(year, Russia_Bluestream,         bottom=Azerbaijan+Iran,                                     width=1, color="blue")    
plt.bar(year, Russia_Turkstream,         bottom=Azerbaijan+Iran+Russia_Bluestream,                   width=1, color="red")                                   
plt.bar(year, Turkmenistan,              bottom=Azerbaijan+Iran+Russia_Bluestream+Russia_Turkstream, width=1, color="black")

plt.text(3.6 ,2  ,label_AZ)
plt.text(3.6 ,11 ,label_IR)
plt.text(3.6 ,30 ,label_RU_BS)
plt.text(3.6 ,33 ,label_RU_TS)
plt.text(3.6 ,37 ,label_TM)

#Horizontal years looked a bit cramped
#plt.xticks(rotation = 45)

#Avoid chopping off labels
plt.tight_layout()

# Save graphic
if lang == "en":
 plt.savefig('gas_contracts_Turkey.svg')
elif lang == "tr":
 plt.savefig("gaz_kontratları_Türkiye.svg")
else:
 print("Unknown language " + lang)
 sys.exit()  
 
# Show graphic
plt.show()

# or if using pythonista
#import dialogs
#from pathlib import Path 
#dialogs.share_url( Path( 'gas_contracts_Turkey.svg').resolve().as_uri())