mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-15 04:22:17 -05:00
0d9fbc1ad7
This version of SickBeard uses both TVDB and TVRage to search and gather it's series data from allowing you to now have access to and download shows that you couldn't before because of being locked into only what TheTVDB had to offer. Also this edition is based off the code we used in our XEM editon so it does come with scene numbering support as well as all the other features our XEM edition has to offer. Please before using this with your existing database (sickbeard.db) please make a backup copy of it and delete any other database files such as cache.db and failed.db if present, we HIGHLY recommend starting out with no database files at all to make this a fresh start but the choice is at your own risk! Enjoy!
100 lines
2.4 KiB
Python
100 lines
2.4 KiB
Python
import gc
|
|
|
|
#---- Default implementation when resource is missing ----------------------
|
|
PAGE_SIZE = 4096
|
|
|
|
def getMemoryLimit():
|
|
"""
|
|
Get current memory limit in bytes.
|
|
|
|
Return None on error.
|
|
"""
|
|
return None
|
|
|
|
def setMemoryLimit(max_mem):
|
|
"""
|
|
Set memory limit in bytes.
|
|
Use value 'None' to disable memory limit.
|
|
|
|
Return True if limit is set, False on error.
|
|
"""
|
|
return False
|
|
|
|
def getMemorySize():
|
|
"""
|
|
Read currenet process memory size: size of available virtual memory.
|
|
This value is NOT the real memory usage.
|
|
|
|
This function only works on Linux (use /proc/self/statm file).
|
|
"""
|
|
try:
|
|
statm = open('/proc/self/statm').readline().split()
|
|
except IOError:
|
|
return None
|
|
return int(statm[0]) * PAGE_SIZE
|
|
|
|
def clearCaches():
|
|
"""
|
|
Try to clear all caches: call gc.collect() (Python garbage collector).
|
|
"""
|
|
gc.collect()
|
|
#import re; re.purge()
|
|
|
|
try:
|
|
#---- 'resource' implementation ---------------------------------------------
|
|
from resource import getpagesize, getrlimit, setrlimit, RLIMIT_AS
|
|
|
|
PAGE_SIZE = getpagesize()
|
|
|
|
def getMemoryLimit():
|
|
try:
|
|
limit = getrlimit(RLIMIT_AS)[0]
|
|
if 0 < limit:
|
|
limit *= PAGE_SIZE
|
|
return limit
|
|
except ValueError:
|
|
return None
|
|
|
|
def setMemoryLimit(max_mem):
|
|
if max_mem is None:
|
|
max_mem = -1
|
|
try:
|
|
setrlimit(RLIMIT_AS, (max_mem, -1))
|
|
return True
|
|
except ValueError:
|
|
return False
|
|
except ImportError:
|
|
pass
|
|
|
|
def limitedMemory(limit, func, *args, **kw):
|
|
"""
|
|
Limit memory grow when calling func(*args, **kw):
|
|
restrict memory grow to 'limit' bytes.
|
|
|
|
Use try/except MemoryError to catch the error.
|
|
"""
|
|
# First step: clear cache to gain memory
|
|
clearCaches()
|
|
|
|
# Get total program size
|
|
max_rss = getMemorySize()
|
|
if max_rss is not None:
|
|
# Get old limit and then set our new memory limit
|
|
old_limit = getMemoryLimit()
|
|
limit = max_rss + limit
|
|
limited = setMemoryLimit(limit)
|
|
else:
|
|
limited = False
|
|
|
|
try:
|
|
# Call function
|
|
return func(*args, **kw)
|
|
finally:
|
|
# and unset our memory limit
|
|
if limited:
|
|
setMemoryLimit(old_limit)
|
|
|
|
# After calling the function: clear all caches
|
|
clearCaches()
|
|
|