def check_in(id):
check_in = request.form.get('check_in')
check_out = request.form.get('check_out')
description = request.form.get('description')
visit = Visit(id, check_in, check_out, description)
VisitRep().save(visit)
return redirect('/pets/' + id + '/visits')
def create_task():
user_id = request.form['user_id']
location_id = request.form['location_id']
review = request.form['review']
user = user_repository.select(user_id)
location = location_repository.select(location_id)
visit = Visit(user, location, review)
visit_repository.save(visit)
return redirect('/visits')
def get_visits(self, pet_id):
visits = []
sql = "SELECT * FROM visits WHERE pet_id = %s"
values = [pet_id]
results = run_sql(sql, values)
for row in results:
visit = Visit(row['pet_id'], row['check_in'], row['check_out'],
row['description'], row['id'])
visits.append(visit)
return visits
def create_trip():
city_id = request.form["city_id"]
country_id = request.form["country_id"]
sight_id = request.form["sight_id"]
to_visit = request.form["to_visit"]
new_city = City("name", city_id)
new_country = Country("country", country_id)
new_sight = Sight("sight", sight_id)
trip = Visit(new_city, new_country, new_sight, to_visit)
visit_repository.save(trip)
return redirect("/trips")
def select_all():
visits = []
sql = "SELECT * FROM visits"
results = run_sql(sql)
for row in results:
user = user_repository.select(row['user_id'])
location = location_repository.select(row['location_id'])
visit = Visit(user, location, row['review'], row['id'])
visits.append(visit)
return visits
def select_all():
visits = []
sql = "SELECT * FROM visits"
results = run_sql(sql)
for row in results:
city = city_repository.select(row['city_id'])
country = country_repository.select(row['country_id'])
sight = sight_repository.select(row["sight_id"])
visit = Visit(city, country, sight, row['to_visit'], row['id'])
visits.append(visit)
return visits
def select(self, id):
visit = None
sql = f"SELECT * FROM {self.table} " + "WHERE id = %s"
values = [id]
result = run_sql(sql, values)[0]
if result is not None:
visit = Visit(
result['pet_id'],
result["check_in"],
result["check_out"],
result['description'],
result["id"],
)
return visit
def select_all(self):
visits = []
sql = f"SELECT * FROM {self.table}"
results = run_sql(sql)
for row in results:
visit = Visit(
row['pet_id'],
row["check_in"],
row["check_out"],
row['description'],
row["id"],
)
visits.append(visit)
return visits
def refresh(self, customer_id):
"""
Refresh customers product list
Args:
customer_id
Returns:
bool
"""
visit = Visit()
selection = ("customer_id", )
filters = [("customer_id", "=")]
values = (customer_id, )
sql = self.q.build("select",
visit.model,
selection=selection,
filters=filters)
success, data = self.q.execute(sql, values=values)
if success:
try:
v_ids = data[0]
sql = "SELECT lines.sku, sum(pcs) AS pcs, product.item " \
"FROM lines " \
"INNER JOIN product ON product.sku = lines.sku " \
"WHERE cust_id IN ? GROUP BY lines.sku"
success, data = self.q.execute(sql, v_ids)
if success:
try:
self._products = [
dict(zip(self.model["fields"], row))
for row in data
]
except IndexError:
pass
return True
except IndexError:
self._products = []
return False
ophth_drugs = Treatment(
'Ophthalmological drugs',
'Oral medications or eye drops can be used to treat infection and other ocular issues, such as cataracts and glaucoma.'
)
treatment7 = TreatmentRep().save(ophth_drugs)
bmt = Treatment(
'Behavioral modification treatments',
'f your animal appears neurotic, obsessive, or overly aggressive and other treatments have not been successful, your veterinarian may prescribe behavioral modifiers like antidepressants or antipsychotics.'
)
treatment8 = TreatmentRep().save(bmt)
##############
# VISITS
visit1 = Visit(pet_1.id, dt.date(2020, 7, 28), dt.date(2020, 7, 30),
'In this visit the pet was so cute the disease cured itself')
VisitRep().save(visit1)
visit2 = Visit(pet_2.id, dt.date(2020, 8, 28), dt.date(2020, 8, 30),
'In this visit the pet was so cute the disease cured itself')
VisitRep().save(visit2)
visit3 = Visit(pet_3.id, dt.date(2020, 9, 28), dt.date(2020, 9, 30),
'In this visit the pet was so cute the disease cured itself')
VisitRep().save(visit3)
visit4 = Visit(pet_4.id, dt.date(2020, 11, 28), dt.date(2020, 12, 30),
'In this visit the pet was so cute the disease cured itself')
VisitRep().save(visit4)
visit5 = Visit(pet_5.id, dt.date(2020, 11, 28), dt.date(2020, 12, 30),
def test_get_visit_by_id_helper(
self,
mock_visit_get_by_id,
mock_user_get_by_pk,
mock_merchant_get_by_pk,
mock_time,
):
test_merchant = Merchant()
test_user = User()
test_visit = Visit()
sess_mock = MagicMock()
test_merchant.pk = 40
test_merchant.merchant_id = 4
test_merchant.merchant_name = "test2"
test_visit.visit_id = uuid.uuid4()
test_visit.timestamp = self.timestamp
test_visit.merchant_pk = 40
test_visit.user_obj_pk = 4
test_request_id = "test2"
test_user.user_obj_id = "20"
test_user.pk = 4
mock_merchant_get_by_pk.return_value = test_merchant
mock_user_get_by_pk.return_value = test_user
mock_visit_get_by_id.return_value = test_visit
test_output = {
"visitId": test_visit.visit_id,
"timestamp": test_visit.timestamp,
"merchant": {
"merchantId": test_merchant.merchant_id,
"merchantName": test_merchant.merchant_name,
},
"user": {
"userId": test_user.user_obj_id
},
}
output = get_visit_by_id_helper(sess_mock, test_request_id,
test_visit.visit_id)
mock_visit_get_by_id.assert_called_once_with(sess_mock,
test_request_id,
test_visit.visit_id)
mock_merchant_get_by_pk.assert_called_once_with(
sess_mock, test_request_id, test_visit.merchant_pk)
mock_user_get_by_pk.assert_called_once_with(sess_mock, test_request_id,
test_visit.user_obj_pk)
mock_time.assert_called_once_with()
self.assertEqual(test_output["merchant"]["merchantId"],
output["merchant"]["merchantId"])
self.assertEqual(test_output["merchant"]["merchantName"],
output["merchant"]["merchantName"])
self.assertEqual(test_output["user"]["userId"],
output["user"]["userId"])
self.assertEqual(type(uuid.uuid4()), type(output["visitId"]))
self.assertEqual(500, output["timestamp"])
user1 = User('Samwise Gamgee')
user_repository.save(user1)
user2 = User('Frodo Baggins')
user_repository.save(user2)
user3 = User('Gollum')
user_repository.save(user3)
location1 = Location('Mordor', 'Attractions')
location_repository.save(location1)
location2 = Location('The Prancing Pony', 'Tavern')
location_repository.save(location2)
visit1 = Visit(user1, location1, '0 stars, far too hot')
visit_repository.save(visit1)
visit2 = Visit(user3, location1, '5 stars, would visit again if I could')
visit_repository.save(visit2)
visit3 = Visit(user1, location2, '4 stars, plenty of beer available')
visit_repository.save(visit3)
visit4 = Visit(user2, location2,
'3 stars, too crowded, could not find my wizard friend')
visit_repository.save(visit4)
pdb.set_trace()
def setUp(self):
self.visit = Visit("Helsinki", "Finland", "lakes")
def makeItem(self):
data = self.getData()
from models.visit import Visit
return Visit(data["start_time"], data["owner"], data["vet"],
data["visit_reason"], data["end_time"],
data["visitanimals"])
def Visit(self, start_time, owner, vet, reason, end_time=None):
return Visit(start_time, owner, vet, reason, end_time)
sight_3 = Sight("Art Gallery")
sight_repository.save(sight_3)
city_1 = Sight("Havana")
city_repository.save(city_1)
city_2 = City("Barcelona")
city_repository.save(city_2)
city_3 = City("Florence")
city_repository.save(city_3)
country_1 = Country("Cuba")
country_repository.save(country_1)
country_2 = Country("Spain")
country_repository.save(country_2)
country_3 = Country("Italy")
country_repository.save(country_3)
visit_1 = Visit(city_2, country_2, sight_2, True)
visit_repository.save(visit_1)
visit_2 = Visit(city_1, country_1, sight_1, True)
visit_repository.save(visit_2)
visit_3 = Visit(city_3, country_3, sight_3, False)
visit_repository.save(visit_3)
pdb.set_trace()
def __init__(self, parent=None):
"""
Initialize MainWindow class
"""
super(MainWindow, self).__init__(parent)
self.setupUi(self)
thread = QThread()
thread.currentThread().setObjectName(__appname__)
configfn.check_config_folder() # Check appdata folder in users home
self.textWorkdate.setText(datetime.date.today().isoformat()
) # initialize workdate to current date
self._archivedOrderlines = OrderLine() # Initialize Detail object
self._archivedVisits = Visit() # Initialize Visit object
self._contacts = Contact() # Initialize Contact object
self._customers = Customer() # Initialize Customer object
self._employees = Employee() # Initialize Employee object
self._orderLines = OrderLine()
self._products = Product() # Initialize Product object
self._reports = Report() # Initialize Report object
self._settings = Settings() # Initialize Settings object
self._visits = Visit()
self.buttonArchiveContacts.clicked.connect(self.archive_contacts)
self.buttonArchiveCustomer.clicked.connect(self.archive_customer)
self.buttonArchiveVisit.clicked.connect(self.archive_visit)
self.buttonCreateContact.clicked.connect(self.create_contact)
self.buttonCreateCustomer.clicked.connect(self.create_customer)
self.buttonCreateReport.clicked.connect(self.create_report)
self.buttonCreateVisit.clicked.connect(self.load_visit)
self.buttonGetCustomers.clicked.connect(self.get_customers)
self.buttonGetPricelist.clicked.connect(self.get_pricelist)
self.toolButtonArchiveSettings.clicked.connect(self.archive_settings)
self.toolButtonCustomer.clicked.connect(self.show_page_customer)
self.toolButtonCustomers.clicked.connect(self.show_page_customers)
self.toolButtonCustomerVisits.clicked.connect(
self.show_page_customer_visits)
self.toolButtonExit.clicked.connect(self.app_exit)
self.toolButtonInfo.clicked.connect(self.show_page_info)
self.toolButtonPricelist.clicked.connect(self.show_page_pricelist)
self.toolButtonReport.clicked.connect(self.show_page_report)
self.toolButtonReports.clicked.connect(self.show_page_reports)
self.toolButtonSettings.clicked.connect(self.show_page_settings)
self.toolButtonCustomerVisit.clicked.connect(self.show_page_visit)
self.toolButtonDeleteSalesData.clicked.connect(self.zero_database)
self.toolButtonExportDatabase.clicked.connect(self.data_export)
self.toolButtonImportCsvData.clicked.connect(
self.show_csv_import_dialog)
self.toolButtonImportDatabase.clicked.connect(self.data_import)
self.widgetCustomers.currentItemChanged.connect(
self.on_customer_changed)
self.widgetCustomers.itemDoubleClicked.connect(
self.on_customer_double_clicked)
self.widgetArchivedVisits.currentItemChanged.connect(
self.on_visit_changed)
self.widgetArchivedVisits.setColumnHidden(0, True)
self.widgetArchivedOrderLines.setColumnWidth(0, 30)
self.widgetArchivedOrderLines.setColumnWidth(1, 30)
self.widgetArchivedOrderLines.setColumnWidth(2, 100)
self.widgetArchivedOrderLines.setColumnWidth(3, 150)
self.widgetArchivedOrderLines.setColumnWidth(4, 60)
self.widgetArchivedOrderLines.setColumnWidth(5, 40)
self.widgetCustomers.setColumnHidden(0, True) # ID
self.widgetCustomers.setColumnWidth(1, 100)
self.widgetCustomers.setColumnWidth(2, 100)
self.widgetCustomers.setColumnWidth(3, 100)
self.widgetCustomers.setColumnWidth(4, 250)
self.widgetCustomers.setColumnWidth(5, 60)
self.widgetPricelist.setColumnWidth(0, 70)
self.widgetPricelist.setColumnWidth(1, 100)
self.widgetPricelist.setColumnWidth(2, 150)
self.widgetPricelist.setColumnWidth(3, 50)
self.widgetPricelist.setColumnWidth(4, 50)
self.widgetPricelist.setColumnWidth(5, 50)
self.widgetPricelist.setColumnWidth(6, 50)
self.widgetPricelist.setColumnWidth(7, 50)
self.widgetPricelist.setColumnWidth(8, 50)
self.widgetPricelist.setColumnWidth(9, 50)
self.widgetPricelist.setColumnWidth(10, 50)
self.widgetPricelist.setColumnWidth(11, 50)
self.widgetPricelist.setColumnWidth(12, 50)
self.widgetReports.setColumnHidden(0, True) # ID
self.widgetReports.setColumnWidth(1, 80) # rep_date
self.widgetReports.setColumnWidth(2, 60) # visits
self.widgetReports.setColumnWidth(3, 60) # sale day
self.widgetReports.setColumnWidth(4, 60) # demo day
self.widgetReports.setColumnWidth(5, 100) # turnover day
self.widgetReports.setColumnWidth(6, 50) # km
# self.widgetReports column 7 # supervisor
self.widgetReportVisits.setColumnWidth(0, 150)
self.widgetReportVisits.setColumnWidth(1, 100)
self.widgetReportVisits.setColumnWidth(2, 100)
self.widgetReportVisits.setColumnWidth(3, 60)
self.populate_customer_list()
self.populate_price_list()
try:
cid = self._settings.settings["cust_idx"]
if self._customers.lookup_by_id(cid):
try:
self.widgetCustomers.setCurrentIndex(
self.widgetCustomers.indexFromItem(
self.widgetCustomers.findItems(str(
self._customers.customer["customer_id"]),
Qt.MatchExactly,
column=0)[0]))
self.toolButtonCustomer.click()
except KeyError:
pass
except KeyError:
return
self._reports.load(workdate=self.textWorkdate.text())
self.populate_report_list()
self.populate_report_visit_list()
self.toolButtonReport.click()
