from candidates.static_data import (
    BaseMapItData, BasePartyData, BaseAreaPostData
)


class MapItData(BaseMapItData):
    pass


class PartyData(BasePartyData):
    def __init__(self):
        super(PartyData, self).__init__()
        self.ALL_PARTY_SETS = (
            {'slug': 'national', 'name': 'National'},
        )

    def party_data_to_party_sets(self, party_data):
        return ['national']


class AreaPostData(BaseAreaPostData):

    def area_to_post_group(self, area_data):
        return None

    def shorten_post_label(self, election, post_label):
        return post_label

    def post_id_to_post_group(self, election, post_id):
        return None

    def post_id_to_party_set(self, post_id):
        return 'national'

from candidates.static_data import (
    BaseMapItData, BasePartyData, BaseAreaPostData
)


class MapItData(BaseMapItData):
    pass


class PartyData(BasePartyData):
    def __init__(self):
        super(PartyData, self).__init__()
        self.ALL_PARTY_SETS = (
            {'slug': 'national', 'name': 'National'},
        )

    def party_data_to_party_sets(self, party_data):
        return ['national']


class AreaPostData(BaseAreaPostData):

    def __init__(self, *args, **kwargs):
        super(AreaPostData, self).__init__(*args, **kwargs)
        self.ALL_POSSIBLE_POST_GROUPS = [None]

    def area_to_post_group(self, area_data):
        return None

    def shorten_post_label(self, election, post_label):
        return post_label

    def post_id_to_post_group(self, election, post_id):
        return None

    def post_id_to_party_set(self, post_id):
        return 'national'

    def party_to_possible_post_groups(self, party_data):
        return (None,)

Fix missing post group defaults for Burkina Faso
Fix missing post group defaults for Burkina Faso

  from candidates.static_data import (
      BaseMapItData, BasePartyData, BaseAreaPostData
  )
  
  
  class MapItData(BaseMapItData):
      pass
  
  
  class PartyData(BasePartyData):
      def __init__(self):
          super(PartyData, self).__init__()
          self.ALL_PARTY_SETS = (
              {'slug': 'national', 'name': 'National'},
          )
  
      def party_data_to_party_sets(self, party_data):
          return ['national']
  
  
  class AreaPostData(BaseAreaPostData):
  
+     def __init__(self, *args, **kwargs):
+         super(AreaPostData, self).__init__(*args, **kwargs)
+         self.ALL_POSSIBLE_POST_GROUPS = [None]
+ 
      def area_to_post_group(self, area_data):
          return None
  
      def shorten_post_label(self, election, post_label):
          return post_label
  
      def post_id_to_post_group(self, election, post_id):
          return None
  
      def post_id_to_party_set(self, post_id):
          return 'national'
  
+     def party_to_possible_post_groups(self, party_data):
+         return (None,)
+ 
Fix missing post group defaults for Burkina Faso
## Code Before:
from candidates.static_data import (
    BaseMapItData, BasePartyData, BaseAreaPostData
)


class MapItData(BaseMapItData):
    pass


class PartyData(BasePartyData):
    def __init__(self):
        super(PartyData, self).__init__()
        self.ALL_PARTY_SETS = (
            {'slug': 'national', 'name': 'National'},
        )

    def party_data_to_party_sets(self, party_data):
        return ['national']


class AreaPostData(BaseAreaPostData):

    def area_to_post_group(self, area_data):
        return None

    def shorten_post_label(self, election, post_label):
        return post_label

    def post_id_to_post_group(self, election, post_id):
        return None

    def post_id_to_party_set(self, post_id):
        return 'national'

## Instruction:
Fix missing post group defaults for Burkina Faso
## Code After:
from candidates.static_data import (
    BaseMapItData, BasePartyData, BaseAreaPostData
)


class MapItData(BaseMapItData):
    pass


class PartyData(BasePartyData):
    def __init__(self):
        super(PartyData, self).__init__()
        self.ALL_PARTY_SETS = (
            {'slug': 'national', 'name': 'National'},
        )

    def party_data_to_party_sets(self, party_data):
        return ['national']


class AreaPostData(BaseAreaPostData):

    def __init__(self, *args, **kwargs):
        super(AreaPostData, self).__init__(*args, **kwargs)
        self.ALL_POSSIBLE_POST_GROUPS = [None]

    def area_to_post_group(self, area_data):
        return None

    def shorten_post_label(self, election, post_label):
        return post_label

    def post_id_to_post_group(self, election, post_id):
        return None

    def post_id_to_party_set(self, post_id):
        return 'national'

    def party_to_possible_post_groups(self, party_data):
        return (None,)
