瀏覽代碼

[changed] Some type hints.

TnS-hun 6 年之前
父節點
當前提交
170d112f1a
共有 2 個文件被更改,包括 4 次插入4 次删除
  1. 1 1
      kobo-book-downloader/KoboDrmRemover.py
  2. 3 3
      kobo-book-downloader/Settings.py

+ 1 - 1
kobo-book-downloader/KoboDrmRemover.py

@@ -13,7 +13,7 @@ class KoboDrmRemover:
 		self.DeviceIdUserIdKey = KoboDrmRemover.__MakeDeviceIdUserIdKey( deviceId, userId )
 
 	@staticmethod
-	def __MakeDeviceIdUserIdKey( deviceId: str, userId: str ) -> str:
+	def __MakeDeviceIdUserIdKey( deviceId: str, userId: str ) -> bytes:
 		deviceIdUserId = ( deviceId + userId ).encode()
 		key = hashlib.sha256( deviceIdUserId ).hexdigest()
 		return binascii.a2b_hex( key[ 32: ] )

+ 3 - 3
kobo-book-downloader/Settings.py

@@ -12,10 +12,10 @@ class Settings:
 
 		self.Load()
 
-	def AreAuthenticationSettingsSet( self ):
-		return len( self.DeviceId) > 0 and len( self.AccessToken ) > 0 and len( self.RefreshToken ) > 0
+	def AreAuthenticationSettingsSet( self ) -> bool:
+		return len( self.DeviceId ) > 0 and len( self.AccessToken ) > 0 and len( self.RefreshToken ) > 0
 
-	def IsLoggedIn( self ):
+	def IsLoggedIn( self ) -> bool:
 		return len( self.UserId ) > 0 and len( self.UserKey ) > 0
 
 	def Load( self ) -> None: