Smart contract changes (#37)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vana"
-version = "0.20.0"
+version = "0.21.0"
 description = ""
 authors = ["Tim Nunamaker <[email protected]>", "Volodymyr Isai <[email protected]>", "Kahtaf Alam <[email protected]>"]
 readme = "README.md"

vana/__init__.py

@@ -15,7 +15,7 @@
 # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 # DEALINGS IN THE SOFTWARE.
 
-__version__ = "0.20.0"
+__version__ = "0.21.0"
 
 import rich
 

vana/client.py

@@ -61,6 +61,18 @@ def get_file(self, file_id: int):
             return None
         return file
 
+    def get_file_permissions(self, file_id: int, account: str) -> str:
+        """
+        Get the permissions for a specific account on a file.
+
+        :param file_id: ID of the file
+        :param account: Address of the account to check permissions for
+        :return: The encryption key for the account, or an empty string if no permissions
+        """
+        get_permissions_fn = self.data_registry_contract.functions.filePermissions(file_id, account)
+        permissions = self.chain_manager.read_contract_fn(get_permissions_fn)
+        return permissions
+
     def add_file(self, url: str, integrity_hash: Optional[str] = None):
         """
         Add a file to the Data Registry contract.
@@ -83,7 +95,7 @@ def get_tee(self, address: str):
         tee = self.chain_manager.read_contract_fn(get_tee_fn)
         if tee is None:
             return None
-        (teeAddress, url, status, amount, withdrawnAmount) = tee
+        (teeAddress, url, status, amount, withdrawnAmount, jobsCount, publicKey) = tee
         if url == "":
             return None
         return tee