Commit 37317ab1 authored by Cristiano Urban's avatar Cristiano Urban
Browse files

Global backend configuration file integration + minor changes.

parent 025d9a43
Loading
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -13,8 +13,8 @@ RUN useradd -m -s /bin/bash curban
# Set up a workdir for the container
WORKDIR /home/transfer_service/

# Copy the source code of the server app
COPY wait-for-it.sh *.py ./
# Copy source code and config file of the server app
COPY wait-for-it.sh *.py vos_ts.conf ./
RUN chown transfer_service wait-for-it.sh *.py && \ 
    chmod 755 wait-for-it.sh *.py        

+7 −1
Original line number Diff line number Diff line
from amqp_server import AMQPServer
from job_cache import JobCache
from config import Config


class AbortJobAMQPServer(AMQPServer):
  
    def __init__(self, host, queue):
        self.type = "abort"
        self.jobCache = JobCache('redis', 6379, 0)
        config = Config("vos_ts.conf")
        self.params = config.loadSection("job_cache")
        self.jobCache = JobCache(self.params["host"], 
                                 self.params.getint("port"), 
                                 self.params.getint("db_read"))
        super(AbortJobAMQPServer, self).__init__(host, queue)      

    def execute_callback(self, requestBody):
+6 −2
Original line number Diff line number Diff line
@@ -2,12 +2,16 @@ import hashlib
import os
import sys

from config import Config


class Checksum(object):

    def __init__(self):
        self.fileBufferSize = 4096
        self.md5FileSuffix = "-md5sum.txt"
        config = Config("vos_ts.conf")
        self.params = config.loadSection("checksum")
        self.fileBufferSize = self.params.getint("file_buffer_size")
        self.md5FileSuffix = self.params["md5_file_suffix"]

    def setFileBufferSize(fileBufferSize):
        self.fileBufferSize = fileBufferSize
+19 −0
Original line number Diff line number Diff line
import configparser


class Config(object):

    def __init__(self, fileName):
        self.fileName = fileName
        self.config = configparser.ConfigParser()
        self.config.read(fileName)        

    def loadSection(self, sectionName):
        section = self.config[sectionName]
        return section

# Test
#c = Config("vos_ts.conf")
#params = c.loadSection("file_grouper")
#print(params.getint("min_num_files"))
#print(eval(params["max_dir_size"]))
+1 −0
Original line number Diff line number Diff line
@@ -3,6 +3,7 @@ import sys

from node import Node


class DbConnector(object):

    def __init__(self, user, password, host, port, dbname):
Loading