Added cookbook support

chef/__init__.py

@@ -6,6 +6,7 @@
 from chef.client import Client
 from chef.data_bag import DataBag, DataBagItem
 from chef.exceptions import ChefError
+from chef.cookbook import Cookbook
 from chef.node import Node
 from chef.role import Role
 from chef.environment import Environment

chef/base.py

@@ -44,12 +44,12 @@ class ChefObject(six.with_metaclass(ChefObjectMeta, object)):
 
     api_version = '0.9'
 
-    def __init__(self, name, api=None, skip_load=False):
+    def __init__(self, name, version='_latest', api=None, skip_load=False):
         self.name = name
         self.api = api or ChefAPI.get_global()
         self._check_api_version(self.api)
+        self.url = self.__class__._build_url(name, version)
 
-        self.url = self.__class__.url + '/' + self.name
         self.exists = False
         data = {}
         if not skip_load:
@@ -138,3 +138,14 @@ def _check_api_version(cls, api):
         # serialization perhaps).
         if api and cls.api_version_parsed > api.version_parsed:
             raise ChefAPIVersionError("Class %s is not compatible with API version %s" % (cls.__name__, api.version))
+
+    @classmethod
+    def _build_url(cls, name, version):
+        """Determine the url for the request"""
+        # Cookbooks need some special handling, and we default to the
+        # latest revision if version goes unspecified.
+        if cls.__name__ == 'Cookbook':
+            url = "{0}/{1}/{2}".format(cls.url, name, version)
+        else:
+            url = cls.url + '/' + name
+        return url

chef/client.py

@@ -12,7 +12,7 @@ def _populate(self, data):
         if self.platform:
             self.orgname = data.get('orgname')
             self.validator = bool(data.get('validator', False))
-            self.public_key = data.get('certificate')
+            self.public_key = data.get('public_key')
             self.admin = False
         else:
             self.admin = bool(data.get('admin', False))
@@ -31,7 +31,7 @@ def to_dict(self):
             d.update({
                 'orgname': self.orgname,
                 'validator': self.validator,
-                'certificate': self.certificate,
+                'public_key': self.certificate,
                 'clientname': self.name,
             })
         else:
@@ -48,6 +48,7 @@ def create(cls, name, api=None, admin=False):
         obj.admin = admin
         d = api.api_request('POST', cls.url, data=obj)
         obj.private_key = d['private_key']
+        obj.public_key = d['public_key']
         return obj
 
     def rekey(self, api=None):

chef/cookbook.py

@@ -0,0 +1,30 @@
+from chef.base import ChefObject, ChefAPI
+from chef.exceptions import ChefServerNotFoundError
+
+class Cookbook(ChefObject):
+    """A Chef Cookbook object."""
+
+    url = '/cookbooks'
+
+    attributes = {
+        'definitions': list,
+        'name': str,
+        'attributes': list,
+        'files': list,
+        'json_class': str,
+        'providers': list,
+        'metadata': dict,
+        'libraries': list,
+        'templates': list,
+        'resources': list,
+        'cookbook_name': str,
+        'version': str,
+        'recipes': list,
+        'root_files': list,
+        'chef_type': str,
+    }
+
+
+    def __getitem__(self, attr):
+         return self.__dict__[attr]
+

chef/tests/test_api.py

@@ -17,7 +17,7 @@ def test_basic(self):
     def test_current_dir(self):
         api = self.load('current_dir.rb')
         path = os.path.join(os.path.dirname(__file__), 'configs', 'test_1')
-        self.assertEqual(api.client, path)
+        self.assertEqual(os.path.normpath(api.client), path)
 
     def test_env_variables(self):
         try:

chef/tests/test_client.py

@@ -15,22 +15,21 @@ def test_get(self):
         self.assertTrue(client.certificate)
         self.assertEqual(client.private_key, None)
 
-    @unittest2.skip('Unknown failure, skipping until tomorrow morning <NPK 2012-03-22>')
     def test_create(self):
         name = self.random()
         client = Client.create(name)
         self.register(client)
         self.assertEqual(client.name, name)
         #self.assertEqual(client.orgname, 'pycheftest') # See CHEF-2019
         self.assertTrue(client.private_key)
-
+        self.assertTrue(client.public_key)
         self.assertIn(name, Client.list())
 
         client2 = Client(name)
         client2.rekey()
+        self.assertEqual(client.public_key, client2.public_key)
         self.assertNotEqual(client.private_key, client2.private_key)
 
-    @unittest2.skip('Unknown failure, skipping until tomorrow morning <NPK 2012-03-22>')
     def test_delete(self):
         name = self.random()
         client = Client.create(name)