import pyodbc


# Based on https://docs.microsoft.com/en-us/azure/sql-database/sql-database-connect-query-python
def connect(user, password, server_uri):
    driver = '{ODBC Driver 13 for SQL Server}'
    if ':' in server_uri:
        server, rest = server_uri.split(':')
        port, database = server_uri.split('/')
    else:
        server, database = server_uri.split('/')
        port = '1443'
    return pyodbc.connect('DRIVER={};PORT={};SERVER={};DATABASE={};UID={};PWD={}'.format(driver, port, server, database, user, password))