From ad66938f122157dc712f2b06a68eb5ada255114a Mon Sep 17 00:00:00 2001
From: witchdocsec <107813117+witchdocsec@users.noreply.github.com>
Date: Wed, 15 May 2024 22:20:03 +0100
Subject: [PATCH] Update presets.py

---
 lib/presets.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/lib/presets.py b/lib/presets.py
index 8d0084a..e637a73 100644
--- a/lib/presets.py
+++ b/lib/presets.py
@@ -16,15 +16,17 @@ class Preset(Base):
 	userfield = Column("userfield", String)
 	passwordfield = Column("passwordfield", String)
 	formnumber = Column("formnumber", Integer)
+	button = Column("button", String)
 	targeturl = Column("targeturl", String)
 
-	def __init__(self,name,browser,url,userfield,passwordfield,formnumber,targeturl):
+	def __init__(self,name,browser,url,userfield,passwordfield,formnumber,button,targeturl):
 		self.name=name
 		self.browser=browser
 		self.url=url
 		self.userfield=userfield
 		self.passwordfield=passwordfield
 		self.formnumber=formnumber
+		self.button=button
 		self.targeturl=targeturl
 
 engine = create_engine("sqlite:///dbs/presets.db")
@@ -33,8 +35,8 @@ Base.metadata.create_all(bind=engine)
 Session = sessionmaker(bind=engine)
 session = Session()
 
-def make_preset(name,browser,url,userfield,passwordfield,formnumber,targeturl):
-	p=Preset(name,browser,url,userfield,passwordfield,int(formnumber),targeturl)
+def make_preset(name,browser,url,userfield,passwordfield,formnumber,button,targeturl):
+	p=Preset(name,browser,url,userfield,passwordfield,int(formnumber),button,targeturl)
 	try:
 		session.add(p)
 		session.commit()
@@ -48,5 +50,6 @@ def load_preset(args):
 	args.userfield = p.userfield
 	args.passwordfield = p.passwordfield
 	args.formnumber = p.formnumber
+	args.button = p.button
 	args.targeturl = p.targeturl
 	return args