thunderstrike/plane
thunderstrike/plane
2
0
Fork 0
forked from github/plane
Code Pull Requests Activity

feat: segregate api token workspace

Browse Source
This commit is contained in:
pablohashescobar 2023-02-06 20:22:08 +05:30
parent e1b7e8d139
commit 0b7f0640c9
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
apiserver/plane/api/views/api_token.py
View File

@ -15,12 +15,16 @@ from plane.api.serializers import APITokenSerializer
class ApiTokenEndpoint(BaseAPIView): class ApiTokenEndpoint(BaseAPIView):
def post(self, request): def post(self, request):
try: try:
label = request.data.get("label", str(uuid4().hex)) label = request.data.get("label", str(uuid4().hex))
workspace = request.data.get("workspace", False)
if not workspace:
return Response(
{"error": "Workspace is required"}, status=status.HTTP_200_OK
)
api_token = APIToken.objects.create( api_token = APIToken.objects.create(
label=label, label=label, user=request.user, workspace=workspace
user=request.user,
) )
serializer = APITokenSerializer(api_token) serializer = APITokenSerializer(api_token)

4
apiserver/plane/db/models/api_token.py
View File

@ -17,7 +17,6 @@ def generate_token():
class APIToken(BaseModel): class APIToken(BaseModel):
token = models.CharField(max_length=255, unique=True, default=generate_token) token = models.CharField(max_length=255, unique=True, default=generate_token)
label = models.CharField(max_length=255, default=generate_label_token) label = models.CharField(max_length=255, default=generate_label_token)
user = models.ForeignKey( user = models.ForeignKey(
@ -28,6 +27,9 @@ class APIToken(BaseModel):
user_type = models.PositiveSmallIntegerField( user_type = models.PositiveSmallIntegerField(
choices=((0, "Human"), (1, "Bot")), default=0 choices=((0, "Human"), (1, "Bot")), default=0
) )
workspace = models.ForeignKey(
"db.workspace", related_name="api_tokens", on_delete=models.SET_NULL, null=True
)
class Meta: class Meta:
verbose_name = "API Token" verbose_name = "API Token"