def get_compnay(id):
session = init_db.init_database()
query = session.query(Company)
query = query.filter(Company.Id == id)
result = query.all()
session.expunge_all()
return result
def insert(self, obj_company: Company):
try:
session = init_db.init_database()
session.add(obj_company)
session.commit()
return True, str(0)
except Exception as ex:
print('error occured!', str(ex))
return False, str(ex)
def update_company(company_id, title, address, phone):
try:
session = init_db.init_database()
query = session.query(Company)
query = query.filter(Company.Id == company_id)
query.update({'Title': title, 'Address': address, 'Phone': phone})
session.commit()
session.close()
return True, str(0)
except Exception as ex:
print('error occurred!', str(ex))
return False, str(ex)
else:
logger.info("Start client in read or write mode.")
## except OSError:
except ConnectionRefusedError:
logger.info("No connection to server.")
except socket.timeout:
logger.info("Timeout is over.")
except KeyboardInterrupt:
logging.debug("The client is closing..")
client_socket.close()
sys.exit()
if __name__ == '__main__':
logger = logging.getLogger('root')
init_database(constants.DB_SERVER)
init_database(constants.DB_CLIENT)
parser = argparse.ArgumentParser(
description='Client. Author: Ryzhova Tanya')
parser.add_argument('addr', help='Server ip address')
parser.add_argument('port', nargs='?', help='TCP-port on server')
parser.add_argument('-r',
'--read_mode',
action='store_const',
const='1',
help='Reading mode')
parser.add_argument('-w',
'--write_mode',
action='store_const',
const='1',
