kevin808 / azure-function-pyodbc-mi Goto Github PK

def get_xml_file(bsoup_obj, source):
        xml = bsoup_obj.find('channel')
        data = {
            "source": source,
            "pubDate": xml.find('pubDate').text,
            "xml_doc": bsoup_obj,
            "xml_raw": bsoup_obj,
            "created": datetime.now()
        }
        return data

def get_sql_connection():
        server = 'sqlsrv.database.windows.net'
        database = 'sqldb'
        driver= '{ODBC Driver 17 for SQL Server}'
        db_token = '<db_token_here>'
        conn_string = 'DRIVER='+driver+';SERVER='+server+';DATABASE='+database
        if os.getenv('MSI_TOKEN'):
            conn = pyodbc.connect(conn_string+';Authentication=ActiveDirectoryMsi')
            return conn
        else:
            SQL_COPT_SS_ACCESS_TOKEN = 1256

            exptoken = b''
            for i in bytes(db_token, 'UTF-8'):
                exptoken += bytes({i})
                exptoken += bytes(1)
            
            tokenstruct = struct.pack('=i', len(exptoken)) + exptoken
            conn = pyodbc.connect(conn_string, attrs_before={SQL_COPT_SS_ACCESS_TOKEN: tokenstruct})
        return conn

def write_sql_data(conn, package):
        cursor = conn.cursor()

        query = cursor.execute(f'INSERT INTO dbo.NEWS_XML (source, pubDate, xml_doc, xml_raw, created) VALUES (?,?,?,?,?)',
        package['source'], package['pubDate'], package['xml_doc'], package['xml_raw'], package['created'])
        conn.commit()

        return query