"""The :class:`Branch <Branch>` object. It holds the information GitHub

# The Accept header will likely be removable once the feature is out of

# preview mode. See: http://git.io/v4O1e

PREVIEW_HEADERS = {'Accept': 'application/vnd.github.loki-preview+json'}

def _update_attributes(self, branch):

#: Name of the branch.

self.name = branch.get('name')

#: Returns the branch's

#: :class:`RepoCommit <github3.repos.commit.RepoCommit>` or ``None``.

self.commit = branch.get('commit')

if self.commit:

self.commit = RepoCommit(self.commit, self)

#: Returns '_links' attribute.

self.links = branch.get('_links', {})

#: Provides the branch's protection status.

self.protection = branch.get('protection')

if 'self' in self.links:

self._api = self.links['self']

else: # Branches obtained via `repo.branches` don't have links.

base = self.commit.url.split('/commit', 1)[0]

self._api = self._build_url('branches', self.name, base_url=base)

def _repr(self):

return '<Repository Branch [{0}]>'.format(self.name)

def latest_sha(self, differs_from=''):

"""Check if SHA-1 is the same as remote branch

# If-None-Match returns 200 instead of 304 value does not have quotes

headers = {

'Accept': 'application/vnd.github.chitauri-preview+sha',

'If-None-Match': '"{0}"'.format(differs_from)

}

base = self._api.split('/branches', 1)[0]

url = self._build_url('commits', self.name, base_url=base)

resp = self._get(url, headers=headers)

if self._boolean(resp, 200, 304):

return resp.content

return None

def protect(self, enforcement=None, status_checks=None):

"""Enable force push protection and configure status check enforcement.

previous_values = self.protection['required_status_checks']

if enforcement is None:

enforcement = previous_values['enforcement_level']

if status_checks is None:

status_checks = previous_values['contexts']

edit = {'protection': {'enabled': True, 'required_status_checks': {

'enforcement_level': enforcement, 'contexts': status_checks}}}

json = self._json(self._patch(self._api, data=dumps(edit),

headers=self.PREVIEW_HEADERS), 200)

self._update_attributes(json)

return True

def unprotect(self):

"""Disable force push protection on this branch."""

edit = {'protection': {'enabled': False}}

json = self._json(self._patch(self._api, data=dumps(edit),

headers=self.PREVIEW_HEADERS), 200)

self._update_attributes(json)