added revoke() method for tokens

Author: masci

oauth2_provider/models.py

@@ -213,6 +213,13 @@ def allow_scopes(self, scopes):
 
         return resource_scopes.issubset(provided_scopes)
 
+    def revoke(self):
+        """
+        Convenience method to uniform tokens' interface, for now
+        simply remove this token from the database in order to revoke it.
+        """
+        self.delete()
+
     def __str__(self):
         return self.token
 
@@ -237,6 +244,13 @@ class RefreshToken(models.Model):
     access_token = models.OneToOneField(AccessToken,
                                         related_name='refresh_token')
 
+    def revoke(self):
+        """
+        Delete this refresh token along with related access token
+        """
+        AccessToken.objects.get(id=self.access_token.id).revoke()
+        self.delete()
+
     def __str__(self):
         return self.token