import requests
import datetime
import hmac
import hashlib
import base64
import pprint
import json
import uuid
import re
import time
from dateutil.parser import parse

storage_account_name = <STORAGE_ACCOUNT_NAME>
storage_account_key = <STORAGE_ACCUNT_KEY>
api_version = '2018-11-09'
request_id = str(uuid.uuid1())
request_time = datetime.datetime.utcnow().strftime('%a, %d %b %Y %H:%M:%S GMT')

#the file path on adls gen2
FILE_SYSTEM_NAME='<FILE_PATH>\nresource:file'

data = "<TEST></TEST>"
content_len = str(len(data))

string_params = {
    'verb': 'PUT',
    'Content-Encoding': '',
    'Content-Language': '',
    'Content-Length': content_len,
    'Content-MD5': '',
    'Content-Type': '', #application/octet-stream',
    'Date': '',
    'If-Modified-Since': '',
    'If-Match': '',
    'If-None-Match': '',
    'If-Unmodified-Since': '',
    'Range': '',
    'CanonicalizedHeaders': f"x-ms-date:{request_time}\nx-ms-version:{api_version}\nx-ms-client-request-id:{request_id}",
    'CanonicalizedResource': f"/{storage_account_name}/{FILE_SYSTEM_NAME}"
    }

string_to_sign = (string_params['verb'] + '\n'
                  + string_params['Content-Encoding'] + '\n'
                  + string_params['Content-Language'] + '\n'
                  + string_params['Content-Length'] + '\n'
                  + string_params['Content-MD5'] + '\n'
                  + string_params['Content-Type'] + '\n'
                  + string_params['Date'] + '\n'
                  + string_params['If-Modified-Since'] + '\n'
                  + string_params['If-Match'] + '\n'
                  + string_params['If-None-Match'] + '\n'
                  + string_params['If-Unmodified-Since'] + '\n'
                  + string_params['Range'] + '\n'
                  + string_params['CanonicalizedHeaders'] + '\n'
                  + string_params['CanonicalizedResource'])

pprint.pprint(string_to_sign)

signed_string = base64.b64encode(
    hmac.new(
        base64.b64decode(storage_account_key), 
        msg=string_to_sign.encode('utf-8'), 
        digestmod=hashlib.sha256).digest()).decode()

params = {
    'resource': 'file'
}

headers = {
    'x-ms-date' : request_time,
    'x-ms-version' : api_version,
    'x-ms-client-request-id': request_id,
    # 'Content-Type': 'application/octet-stream',
    # 'Content-Length': content_len,
    'Authorization' : f'SharedKey {storage_account_name}:{signed_string}'
}

pprint.pprint(string_params)
print(headers)
url = ('https://' + storage_account_name + '.dfs.core.windows.net/'+'<FILE_PATH>')
r = requests.put(url, headers=headers, params=params, data=data)

#print out the file content
print(r.content)