Browse Source

new feature

unknown 4 years ago
parent
commit
e8c241daae
3 changed files with 26 additions and 0 deletions
  1. 2 0
      cartridges.gohtml
  2. 12 0
      generate.html
  3. 12 0
      main.go

+ 2 - 0
cartridges.gohtml

@@ -10,5 +10,7 @@
 		<li>{{ . }}</li>
 		{{ end }}
 	</ul>
+
+	<input type="button" name="UpdateCartridgesBtn" value="Назад" onclick="location.href = '/'">
 </body>
 </html>

+ 12 - 0
generate.html

@@ -43,6 +43,18 @@
 		<input type="submit" value="Добавить совместимость">
 	</form>
 	<br><br>
+
+	<p>Добавление картриджей</p>
+	<form name="addcartridge" action="/updateCartridge">
+		<select name="cartridges">
+			<option></option>
+			{{ range .CartridgeNames }}
+			<option value="{{ . }}"></option>
+			{{ end }}
+			<input type="number" name="cartridgeQuantity">
+		</select>
+	</form>
+	<br><br>
 	<input type="button" value="Найти совместимый картридж" onclick="location.href = '/compatible'">
 	<input type="button" value="Состояние картриджей" onclick="location.href = '/cartridges'">
 </body>

+ 12 - 0
main.go

@@ -38,12 +38,24 @@ func main() {
 	r.HandleFunc("/compatible", FindCompatibleCartridges)
 	r.HandleFunc("/printer/{printerName}", PrinterPage)
 	r.HandleFunc("/cartridges", CartridgePage)
+	r.HandleFunc("/updateCartridge", updateCartridge)
 	fmt.Println("Server is listening...")
 
 	http.Handle("/", r)
 	http.ListenAndServe(":"+port, nil)
 }
 
+func updateCartridges(w http.ResponseWriter, r *http.Request){
+	db, _ := gorm.Open(sqlite.Open("printer.db"), &gorm.Config{})
+
+	r.ParseForm()
+	selectedCartridge := strings.Join(r.Form["cartridges"], "")
+	var cartridge Cartridges
+	db.Where("Name = ?", selectedCartridge).First(&cartridge)
+	newQuantity := cartridge.Quantity + r.Form["cartridgeQuantity"]
+	db.Model(&cartridge).Update("Quantity", newQuantity)
+}
+
 func generateCompatible(w http.ResponseWriter, r *http.Request) {
 	fmt.Println("New generating request from" + r.RemoteAddr)
 	data, err := ioutil.ReadAll(r.Body)