From ad0d631e346768f4dfc671d722ea66c90829aa40 Mon Sep 17 00:00:00 2001
From: TanmayArya <72544600+TanmayArya-1p@users.noreply.github.com>
Date: Mon, 16 Dec 2024 19:40:49 +0530
Subject: [PATCH] checking if keyloaded

---
 __pycache__/manager.cpython-312.pyc | Bin 0 -> 3439 bytes
 main.py                             |  14 ++++++++++----
 manager.py                          |   3 +++
 3 files changed, 13 insertions(+), 4 deletions(-)
 create mode 100644 __pycache__/manager.cpython-312.pyc

diff --git a/__pycache__/manager.cpython-312.pyc b/__pycache__/manager.cpython-312.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..31cec5520f142f12d6aff4f6ff0d95d7889f7831
GIT binary patch
literal 3439
zcmbtXO>7fK6rS0i*iP)&!677IgJVJ(V;Tqq3KUh%kAO;Tgi88Dl(KBSo8Z*3(^-e$
z$VeceDpgZ8Q7Z+hsuWbI1aX1(z_sPrUgDa{)+&`M_0(HP4v})|d%Ir8f1DUAZ)fJc
znb~>od*9pq;d0dzXqS7hjjxG>{EGvt35CvP2XvN*LKGS!lXQeq!jTI^;qDTJf5e$q
zBV0-diT~jsMD@D!DvPVhh^aGW!H{g8hVc?n$p}@*2&d4yWQ51fMg)ZiDJl*{yh}&w
zlsXvM6dOppVh8DfBkFbcC0WyE5=?ndj?3dJi_nr&@DLvUHqRM{C?rj!qS_c0t)r;3
zUJC98BV1{;6bW%d=R<<df2O8%*Hqy|X)GF3ONL4`lGGhw5=+R6s=&nsO^uCdct&9R
z%)r^v4>Xl&qr-B1QXU;lM5ZUzcv2f3NGQ?x`1|TqLL0p_Q@F?Iq<Nq2sg!P)q-Z>v
zl%(0#iraQunmRZ`3xK#y^4>jj!n{52Z=VyM+xO=E2h3;N{yE|6#!avxk>W*RfJ+T0
z{SKWak}SGpqSOj>*dCdWGG`@ig-Md83bi0`lHB6b+=Ro@?L5<D!UEIIBzaUgmm;*_
zXYB)H4uS!2G+T-sBt1l<GaMyX2!&sz{w9Bg%+L@&q;oT$4u|R(-cEOpt8tadNmYUf
zF??lRNKC15U7TUjq{<Na4|&#LD5m7(b)AlBxH1q3FelE|N0?ezKfAXw5IZ`w)!qeh
zo2+@;a^B92w=?VQ!kDbNnie|l30aqaolwXA<)*yT_05GZFDwi$U0l5Q-09!qVfvpo
zKSpXx-05#?_+CHx`gZZZ?GpQfw%QS#2cyym)`&+D)L>d@wPIQ`2F0~6i9w0JBnAv(
zAh2C<m@a?=6~p(2yIC_#miyR_xq}g^hPtJ}owPF`z`tew41hoPv@YFPyz%YLxxx91
z=*&QgGp+u%hM@eq<plqW*vH$3BE|QHX~Y?hH)9}{0MVtCk>>7%0n?@SV1>={;UM{B
z4<VzFIk)&*0wBGJThk6@nokQm=$>yAND6)r=}?GgcF^l0;BivZowA}x7TJVE1Db3%
z8Z;suMCF)tLyk?W8t@FP(CwD3hKB-dE#Nj6z-j3X;+pMv353h6;4(z(0|C5Z5lTbb
zYQv#?@K`Q*Dib`F4W8a0)OCEZaY0zPp7%88Jnb1z`*O#M{Vz|?3m#PKJaKliM%bk`
zY&#+KD_xx|I=LFe;N%E8IL+5`ufkbWVuV4sC6*8rnLrO$G|PShldPhp#R*r%6b!9!
z7gxGpZUZ3jRBO`$)}(n=AdV`+t4^;J*-0qMw&~fB!+?sJR-!f)izW@4Gl|Wns1cT$
zO6qlLJd#jUy$&Ye$q@KJh<Hx}Nvs8Wx)6&3HMV2d0Sj=Ys%Tgj3=kFQZ<mpPW$2g<
zR${2y#t++QUx3&_keRmpSJFD+9Wq^}`PQ}#!ntI+AmoFc4-VfyymI-+ql<1pRCF!e
z^SI+_*ORVn_|05+AQK+=^_|?{aAt5gJNRKX{827^H50y?4PRRja-Lwu6I|~7)6?}r
z#O=0t+<(hB<%J=)HKgb7`a=CS{+Uhe_t?s^AIpM~{coF)gvv7kQ<W`EF?gV>Q6cZl
zB#Sv;+oG&S@~;F=?p47F9K}yGRo)NoM!(0pDuCpF5G<Hi*Jfm{%6eP`t1vXn$zj$B
z+i%l)fGomq2e+kwEx?Y2(Bd!(F9?hD5SlEA0`ba@fT-X&b^zvPeU)%5WzY^>wFm-$
zcrZLC<m`Jh_PzIdGxk8f!JTXHXBzyg4WYHB-3v;tr7P3YwbGGoI=T+P?0MX?<^y0o
zIDP+g)^{Z5>&^IjAAkH*dLm^{p3nL&<a|RJ-%!?fX~B{AG%eW{ZTEWbbuQE8(DJ^O
z{?*o_&ppSs0Fph%Zpep+w>WEPV)Ey{*8XPxS+m&hFR$(byt0AGv(N#Wtcl24HK~+1
zFXJcdtk?2SAf8AD#uC$Ur8~qKVbS?<H3@x#8qnVwV%w@9@bQ!^mZEAmmukbf)&~OI
z?Dpo|?HPA_&V4B3KD4Z?9_h=v`&aG#MloPW4e$vMpSN*&Qk5jVUXmsg%5)6-PD%QF
zT8^0-h8)bW7_ogQ4x)G+1;(A70P&ERn__-5%s@k+^R(AM+$9@=ARgFo3gZ3^Cok^b
ztQW-N8%|M_=|-a{_G~s%ae$h8;Bu(Oy4T>s#5j|uuBW=kjQ?Ux)EWo8fu~?uGrkYE
k?dQywe~aPQ2@nX;@B?R^r<DFpI=AY&=m6ayD2-MB17hg3umAu6

literal 0
HcmV?d00001

diff --git a/main.py b/main.py
index 57ab52b..32d0f26 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,12 @@
 from manager import PasswordManager
 
 
+def validate_key_loaded(pm : PasswordManager):
+    if not pm.keyloaded:
+        print("Key not loaded. Please load a key first.")
+        return False
+    return True
+
 def main():
     password = {
         "gmail": "password1",
@@ -29,17 +35,17 @@ def main():
         elif choice == '2':
             path = input("Enter key file path: ").strip()
             pm.load_key(path)
-        elif choice == '3':
+        elif choice == '3' and validate_key_loaded(pm):
             path = input("Enter password file path: ").strip()
             pm.create_password_file(path, password)
-        elif choice == '4':
+        elif choice == '4' and validate_key_loaded(pm):
             path = input("Enter password file path: ").strip()
             pm.load_password_file(path)
-        elif choice == '5':
+        elif choice == '5' and validate_key_loaded(pm):
             site = input("Enter site: ").strip()
             password = input("Enter password: ").strip()
             pm.add_password(site, password)
-        elif choice == '6':
+        elif choice == '6' and validate_key_loaded(pm):
             site = input("Enter site: ").strip()
             print(f"Password for {site}: {pm.get_password(site)}")
         elif choice == 'q':
diff --git a/manager.py b/manager.py
index 212f675..fddb533 100644
--- a/manager.py
+++ b/manager.py
@@ -7,15 +7,18 @@ def __init__(self):
         self.key = None
         self.password_file = None
         self.password_dict = {}
+        self.keyloaded = False
 
     def create_key(self, path):
         self.key = Fernet.generate_key()
         with open(path, 'wb') as f:
             f.write(self.key)
+        self.keyloaded = True
 
     def load_key(self, path):
         with open(path, 'rb') as f:
             self.key = f.read()
+        self.keyloaded = True
 
     def create_password_file(self, path, initial_values=None):
         self.password_file = path