oci/zeronet
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Auto-pin downloaded bigger files

Browse source
This commit is contained in:
shortcutme 2017-10-04 13:13:00 +02:00
parent 6e55c8b7c7
commit cd025316db
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE
1 changed files with 16 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
plugins/OptionalManager/OptionalManagerPlugin.py
View file

@ -54,11 +54,22 @@ if "access_log" not in locals().keys(): # To keep between module reloads
@PluginManager.registerTo("WorkerManager") @PluginManager.registerTo("WorkerManager")
class WorkerManagerPlugin(object): class WorkerManagerPlugin(object):
def doneTask(self, task): def doneTask(self, task):
if task["optional_hash_id"]: content_db = self.site.content_manager.contents.db
content_db = self.site.content_manager.contents.db if task["optional_hash_id"] and task["optional_hash_id"] not in self.site.content_manager.hashfield:
inner_path = task["inner_path"]
is_pinned = 0
if "|" in inner_path: # Big file piece
inner_path, file_range = inner_path.split("|")
file_info = self.site.content_manager.getFileInfo(inner_path)
# Auto-pin bigfiles
if config.pin_bigfile and file_info["size"] > 1024 * 1024 * config.pin_bigfile:
is_pinned = 1
content_db.executeDelayed( content_db.executeDelayed(
"UPDATE file_optional SET time_downloaded = :now, is_downloaded = 1, peer = peer + 1 WHERE site_id = :site_id AND inner_path = :inner_path", "UPDATE file_optional SET time_downloaded = :now, is_downloaded = 1, peer = peer + 1, is_pinned = :is_pinned WHERE site_id = :site_id AND inner_path = :inner_path",
{"now": int(time.time()), "site_id": content_db.site_ids[self.site.address], "inner_path": task["inner_path"]} {"now": int(time.time()), "site_id": content_db.site_ids[self.site.address], "inner_path": inner_path, "is_pinned": is_pinned}
) )
super(WorkerManagerPlugin, self).doneTask(task) super(WorkerManagerPlugin, self).doneTask(task)
@ -120,5 +131,6 @@ class ConfigPlugin(object):
def createArguments(self): def createArguments(self):
group = self.parser.add_argument_group("OptionalManager plugin") group = self.parser.add_argument_group("OptionalManager plugin")
group.add_argument('--optional_limit', help='Limit total size of optional files', default="10%", metavar="GB or free space %") group.add_argument('--optional_limit', help='Limit total size of optional files', default="10%", metavar="GB or free space %")
group.add_argument('--pin_bigfile', help='Automatically pin files larger than this limit', default=20, metavar="MB", type=int)
return super(ConfigPlugin, self).createArguments() return super(ConfigPlugin, self).createArguments()