spacepaste

Paste details

reply | raw

  1.  
  2. import os
  4. VersionNum = 'ALPHA'
  7. def GetDirectorySize(directory):
  8. """
  9. Returns interger / float
  10. Returns the size (in btyes) of a given directory or file.
  11. """
  13. tmpTotalBytez = 0 #Variable to store culumative size of directory.
  15. #check if directory is in fact a directory (and not a file).
  16. if os.path.isdir(directory):
  18. #walk through directory and each sub directory and add size of each file.
  19. for root, dirs, files in os.walk(directory):
  20. for file in files:
  21. try:
  22. tmpTotalBytez += os.path.getsize(root + '\\' + file)
  23. except:
  24. pass #Probably because of access denied: TODO: implement a way to track these.
  26. #check is directory is a file.
  27. elif os.path.isfile(directory):
  28. tmpTotalBytez = os.path.getsize(directory)
  30. return tmpTotalBytez
  32. def ClearScreen():
  33. """
  34. Clears the whole console screen
  35. """
  37. if os.name == "posix":
  38. # Unix/Linux/MacOS/BSD/etc
  39. os.system('clear')
  40. elif os.name in ("nt", "dos", "ce"):
  41. # DOS/Windows
  42. os.system('CLS')
  43. else:
  44. # Fallback for other operating systems.
  45. print ('\n' * 100)
  49. def UpOneLevel(directory):
  50. """
  51. Retians a string
  52. Removes the last folder from a string containing a path (nt only)
  53. """
  55. tail = ''
  56. while tail == '': #moves last directory
  57. directory, tail = os.path.split(directory)
  58. if directory.endswith(':\\'): break #if this is the root of Windows drive break loop
  60. if directory.endswith('\\') == False: #ensures '\\' is added incase it is not
  61. directory = directory + '\\'
  62. return directory
  63. def GetDirectorySizes(directory, Verbose = False):
  64. """
  65. Returns a list.
  66. list contains a list of the name and size (in bytes) of each object in the directory.
  67. [[objectName,objectSize]].
  68. Pass Verbose = True to enable some console prints of progress.
  69. """
  71. # create a fresh list to store name and size of each folder and file
  72. FolderSizeList = []
  73. TotalRootSize = 0
  75. CurrentFile = 0
  77. # grab objects in current directory
  78. if Verbose == True:
  79. tmpNumOfDirs = len(os.listdir(directory))
  80. tmpCurrentDir = 1
  82. for RootDirs in os.listdir(directory):
  84. if Verbose == True:
  85. ClearScreen()
  86. print('Currently calculating ' + RootDirs + ' (' + str(tmpCurrentDir) + '/' + str(tmpNumOfDirs) + ')' )
  87. tmpCurrentDir += 1
  89. tmp = GetDirectorySize(directory + '\\' + RootDirs)
  90. FolderSizeList.append([RootDirs, tmp])
  94. return FolderSizeList
  96. def MainMenu(directory):
  98. """
  99. This will present a main screen
  100. show directories of /path
  101. Continues presenting directory until user states a directory to use.
  102. """
  104. OldDirectory = directory #back up incase moving into a directory fails (due to access denied or something)
  105. while True:
  109. #Main Menu
  110. ClearScreen()
  111. print ('Welcome to DirectorySizes')
  112. print ('Version: ' + VersionNum)
  113. print ()
  114. print ('Current Directory: ' + directory)
  116. Indices = 0 # used to present options
  117. OptionList = [] #create a blank list to store options
  119. #stops runtime error on access denied.
  120. try:
  121. SubDirList = os.listdir(directory)
  122. except:
  123. input('Access Denied (5): Returning to previous directory, Press any key')
  124. directory = OldDirectory
  125. SubDirList = os.listdir(directory)
  126. print ('Welcome to DirectorySizes')
  127. print ('Version: ' + VersionNum)
  128. print ()
  129. print ('Current Directory: ' + directory)
  132. for Directory in SubDirList:
  133. if os.path.isdir(directory + Directory):
  134. printline = '{0:4}' + (' ' * 2) + '{1:24}'
  135. print(printline.format(str(Indices) + ')', Directory)) #Prints each folder in the current directory
  136. Indices += 1
  137. OptionList.append(Directory)
  138. #Option for going up one level
  139. OptionList.append('Up one level')
  140. printline = '{0:4}' + (' ' * 2) + '{1:24}'
  141. print(printline.format(str(Indices) + ')','Up one level'))
  142. Indices += 1
  144. #Option Use Current
  145. OptionList.append('Use current directory')
  146. printline = '{0:4}' + (' ' * 2) + '{1:24}'
  147. print(printline.format(str(Indices) + ')','Use current directory'))
  148. print
  150. #grab user input
  152. SelectedOption = int(input('Select an option; '))
  154. #return a value
  156. if OptionList[SelectedOption] == 'Up one level':
  157. OldDirectory = directory
  158. directory = UpOneLevel(directory)
  159. #loop code
  162. elif OptionList[SelectedOption] == 'Use current directory':
  163. #Complete = True
  164. return directory
  167. else:
  168. OldDirectory = directory
  169. directory = directory + OptionList[SelectedOption] + '\\'
  170. #loop code
  172. def ConvertByteSize(SizeInBytes, form = 'Standard' ):
  173. """
  174. Converts an amount of bytes, given as an interger, into KB, MB etc...
  175. returns a list [value(int/float), format(string)]
  176. Will convert based on algorythm or to a format given as form.
  177. Options for form are:
  178. Standard,
  179. B,
  180. KB,
  181. MB,
  182. GB.
  183. """
  184. tmpTotalBytez = [0,'']
  185. tmpTotalBytez[0] = SizeInBytes
  187. #Based on 'form' paramater, select a format to give back the data
  188. if form == 'GB':
  189. tmpTotalBytez[0] = tmpTotalBytez[0]/1024/1024/1024 # convert to GB
  190. tmpTotalBytez[1] = 'GB'
  191. return tmpTotalBytez
  193. elif form == 'MB':
  194. tmpTotalBytez[0] = tmpTotalBytez[0]/1024/1024 # convert to MB
  195. tmpTotalBytez[1] = 'MB'
  196. return tmpTotalBytez
  198. elif form == 'KB':
  199. tmpTotalBytez[0] = tmpTotalBytez[0]/1024 # convert to KB
  200. tmpTotalBytez[1] = 'KB'
  201. return tmpTotalBytez
  203. elif form == 'B':
  204. tmpTotalBytez[1] = 'B'
  205. return tmpTotalBytez
  207. else:
  209. if tmpTotalBytez[0] > 1024*1024*1024*1.3:
  210. tmpTotalBytez[0] = tmpTotalBytez[0]/1024/1024/1024 # convert to GB
  211. tmpTotalBytez[1] = 'GB'
  212. return tmpTotalBytez
  214. elif tmpTotalBytez[0] > 1024*1024*1.3:
  215. tmpTotalBytez[0] = tmpTotalBytez[0]/1024/1024 # convert to MB
  216. tmpTotalBytez[1] = 'MB'
  217. return tmpTotalBytez
  219. elif tmpTotalBytez[0] > 1024*1.3:
  220. tmpTotalBytez[0] = tmpTotalBytez[0]/1024 # convert to KB
  221. tmpTotalBytez[1] = 'KB'
  222. return tmpTotalBytez
  224. else:
  225. tmpTotalBytez[1] = 'B'
  226. return tmpTotalBytez
  227.  
Powered by spacepaste. Paste removal.