first try with archive dept.

workspace
Brendan Howell 3 years ago
parent 82ec144985
commit 427d7a21cd

@ -3,6 +3,10 @@ import os
from bureau import Bureau, add_command from bureau import Bureau, add_command
import pyudev import pyudev
AUDIO_FILETYPES = (".mp3", ".ogg", ".oga", ".flac", ".spx", ".opus", ".wav",
".aac", ".mka", ".m4a")
PICTURE_FILETYPES = (".jpg", ".jpeg", ".png", ".gif", ".tif", ".bmp")
PDF_FILETYPES = (".pdf", ".djvu", ".djv", ".ps", ".eps")
class Archives(Bureau): class Archives(Bureau):
""" """
@ -18,25 +22,112 @@ class Archives(Bureau):
def __init__(self): def __init__(self):
Bureau.__init__(self) Bureau.__init__(self)
# get location of archive root storage from config # TODO: get location of archive root storage from config
self.mounts = {}
# create a db to map shortcodes to folders on external drives
self.dirmap = self.open_db("dirmap")
def __scandir(self, dir_to_scan):
"""
recursive directory scan
returns a tuple of lists of directory paths if any interesting media
are found
([dirs_with_audio], [dirs_with_pictures], [dirs_with_pdfs])
"""
dirs_with_audio = []
dirs_with_pictures = []
dirs_with_pdfs = []
for root, dirs, files in os.walk(dir_to_scan):
audio_found = False
pic_found = False
pdf_found = False
for file in files:
_, file_ext = os.path.splitext(file)
file_ext = file_ext.lower()
if file_ext in AUDIO_FILETYPES:
audio_found = True
elif file_ext in PICTURE_FILETYPES:
pic_found = True
elif file_ext in PDF_FILETYPES:
pdf_found = True
if audio_found:
dirs_with_audio.append(root)
if pic_found:
dirs_with_pictures.append(root)
if pdf_found:
dirs_with_pdfs.append(root)
return dirs_with_audio, dirs_with_pictures, dirs_with_pdfs
def _mount_and_archive(self, dev): def _mount_and_archive(self, dev):
# TODO: small print options to import stuff # TODO: small print options to import stuff
print("MOUNTING EXTERNAL MEDIUM: ", dev) mountpoint = os.path.join("/media" os.path.split(dev))
print("MOUNTING EXTERNAL MEDIUM: ", dev, "on", mountpoint)
if len(self.mounts < 26):
mount_index = chr(65 + len(self.mounts))
# pmount dev # pmount dev
os.system("pmount " + str(dev))
# path is always /media/device # path is always /media/device
os.system("pmount " + str(dev))
#os.walk() self.mounts[mount_index] = {"dev": dev, "mount": mountpoint, "dirs": []}
prn = self._get_small_printer()
# walk through the tree looking for mp3 / ogg prn.textln("Achive has found external medium: ")
prn.textln(mountpoint)
# walk through looking for photos prn.textln("EJECT DRIVE:")
prn.soft_barcode("code128", "ARej." + mount_index, module_width=0.16)
# walk through looking for pdfs
audio_dirs, pic_dirs, pdf_dirs = self.__scandir(mountpoint)
# pumount dev
# small_print list of media/directories that we can import
# TODO: make short codes for each dir
# TODO: make command that will import the dir
for adir in audio_dirs:
print("could add audio DIR", adir)
prn.textln("Import Audio Folder:")
prn.textln(adir)
shortcode = self.dirmap.store_and_get_shortcode(adir)
prn.soft_barcode("code128", "ARia." + shortcode, module_width=0.16)
self.mounts[mount_index]["dirs"].append(shortcode)
for pdir in pic_dirs:
print("could add picture DIR", pdir)
prn.textln("Import Image Folder:")
prn.textln(pdir)
shortcode = self.dirmap.store_and_get_shortcode(pdir)
prn.soft_barcode("code128", "ARii." + shortcode, module_width=0.16)
self.mounts[mount_index]["dirs"].append(shortcode)
for pdir in pdf_dirs:
print("could add document DIR", pdir)
prn.textln("Import Document Folder:")
prn.textln(pdir)
shortcode = self.dirmap.store_and_get_shortcode(pdir)
prn.soft_barcode("code128", "ARip." + shortcode, module_width=0.16)
self.mounts[mount_index]["dirs"].append(shortcode)
self._free_small_printer(prn)
@add_command("ej", "Eject Drive")
def unmount_device(self, data):
"""
Safely eject external media like USB Flash Drives or SD Cards
"""
device_idx, _ = data.split(".")
dev = self.mounts[device_idx]["dev"]
os.system("pumount " + str(dev))
# clean up the references
for shortcode in self.mounts[device_idx]["dirs"]:
self.dirmap.delete(shortcode)
del(mounts[device_idx])
def import_audio(self, data):
"""
Import audio files from external media into the
"""
shortcode, _ = data.split(".")
indir = self.dirmap.get(shortcode)
print("Importing audio:", indir)
print("NOT IMPLEMENTED YET")
def run_io(self): def run_io(self):
# monitor for insert of any removable block storage devices (SD/USB) # monitor for insert of any removable block storage devices (SD/USB)
@ -48,7 +139,6 @@ class Archives(Bureau):
self._mount_and_archive(device.device_node) self._mount_and_archive(device.device_node)
def main(): def main():
ar = Archives() ar = Archives()
ar.run() ar.run()

Loading…
Cancel
Save