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',