mirror of
https://github.com/ada-dmitry/sun_project.git
synced 2026-09-24 08:40:14 +00:00
121 lines
4.5 KiB
Python
121 lines
4.5 KiB
Python
from dbtable import *
|
|
import add_func
|
|
|
|
class StationTable(DbTable):
|
|
'''
|
|
Это класс для таблиц
|
|
'''
|
|
def table_name(self):
|
|
'''
|
|
Это метод для названия таблицы
|
|
'''
|
|
return self.dbconn.prefix + "stations"
|
|
|
|
def columns(self):
|
|
return {"id": ["serial", "PRIMARY KEY"],
|
|
"st_name": ["varchar(20)", "NOT NULL"],
|
|
"tarrif_zone_id": ["integer", "NOT NULL"],
|
|
"st_index": ["integer", "NOT NULL"]}
|
|
|
|
def table_constraint(self):
|
|
return ['CONTRAINT "Name" UNIQUE (st_name)']
|
|
|
|
def find_by_position(self, num):
|
|
sql = "SELECT * FROM " + self.table_name()
|
|
sql += " ORDER BY "
|
|
sql += ", ".join(self.primary_key())
|
|
sql += " LIMIT 1 OFFSET %(offset)s"
|
|
cur = self.dbconn.conn.cursor()
|
|
cur.execute(sql, {"offset": num - 1})
|
|
return cur.fetchone()
|
|
|
|
def delete(self, val):
|
|
par_sql = f"DELETE FROM stations WHERE st_name = '{val[1]}';"
|
|
cur = self.dbconn.conn.cursor()
|
|
cur.execute(par_sql)
|
|
self.dbconn.conn.commit()
|
|
|
|
def insert(self):
|
|
while True:
|
|
st_name = input('Введите название добавляемой станции (enter - отмена): ').strip()
|
|
if len(st_name) > 20:
|
|
print('Недопустмая длина названия станции')
|
|
elif st_name == '':
|
|
return "1"
|
|
else:
|
|
break
|
|
tarrif_zone_id = add_func.validate_input('Введите номер тарифной зоны: ', 0, 7)
|
|
if(tarrif_zone_id==-1):
|
|
return "1"
|
|
st_index = add_func.validate_input('Введите индекс добавляемой станции: ', 0, 100)
|
|
if(st_index==-1):
|
|
return "1"
|
|
insert = [st_index, st_name, tarrif_zone_id]
|
|
self.insert_one(insert)
|
|
|
|
|
|
def update(self, old):
|
|
"""Функция для обновления станции
|
|
"""
|
|
print(f'''Выбрана станция для изменения: {old[1]}''')
|
|
cur = self.dbconn.conn.cursor()
|
|
while True:
|
|
data = input("Введите название (enter - отмена): ").strip()
|
|
if(len(data.strip()) > 20):
|
|
data = input("Название слишком длинное! Введите название заново (enter - отмена):").strip()
|
|
elif(self.check_by_name(data.strip())):
|
|
data = input("Такое название уже существует. Введите новое (enter - отмена):").strip()
|
|
elif(data==''): return
|
|
else:
|
|
param_sql = f"UPDATE {self.table_name()} SET st_name = '{data}' WHERE st_name = '{old[1]}';"
|
|
cur.execute(param_sql)
|
|
self.dbconn.conn.commit()
|
|
return
|
|
|
|
def check_by_name(self, value):
|
|
sql = f"SELECT * FROM {self.table_name()} WHERE st_name='{value}'"
|
|
cur = self.dbconn.conn.cursor()
|
|
cur.execute(sql)
|
|
result = cur.fetchone()
|
|
cur.close()
|
|
if result:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def find_by_name(self, name):
|
|
cur = self.dbconn.conn.cursor()
|
|
param_query = f"SELECT id FROM Stations WHERE st_name = '{name}';"
|
|
# sql_sel = "SELECT id FROM " + self.table_name()
|
|
# sql_sel += " WHERE cath_name = " + "'" + name + "'" + ";"
|
|
cur.execute(param_query)
|
|
ret = cur.fetchone()
|
|
return ret
|
|
|
|
def name_by_id(self, id):
|
|
cur = self.dbconn.conn.cursor()
|
|
query = "SELECT st_name FROM Stations WHERE id = %s;"
|
|
cur.execute(query, str(id))
|
|
return cur.fetchone()
|
|
|
|
def index_by_id(self,id):
|
|
cur = self.dbconn.conn.cursor()
|
|
query = f"SELECT st_index FROM Stations WHERE id = {id};"
|
|
cur.execute(query)
|
|
return cur.fetchone()
|
|
|
|
def example_insert(self):
|
|
self.insert_one([1, "Нахабино", 1])
|
|
self.insert_one([2, "Стрешнево", 2])
|
|
self.insert_one([2, "Царицыно", 2])
|
|
self.insert_one([3, "Яуза", 3])
|
|
self.insert_one([2, "Щукинская", 3])
|
|
self.insert_one([4, "Москва-Курская", 4])
|
|
self.insert_one([4, "Дмитровская", 4])
|
|
self.insert_one([4, "Остафьево", 4])
|
|
self.insert_one([4, "Подольск", 4])
|
|
return
|
|
|
|
|
|
|