3
� �f � @ s, d dl Z d dlZd dlZG dd� de�ZdS )� Nc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�PickleDatabasez� This object is a wrapper to access and handle our
local pickle file. We'll use this pickle file as a sort of
database for storing schedules and other info for the agent.c C sx || _ ytjt|d��| _W nV tk
rr tj| jj �}|j
dj | j �� i | _t| j d� |j d| j � Y nX d S )N�rbz1Unable to open database {}, creating an empty one�wbzDatabase created: %s)
� file_path�pickle�load�open�data� Exception�logging� getLogger� __class__�__name__�warn�format�info)�selfr �log� r �,/usr/lib/fm-agent/library/pickle_database.py�__init__
s zPickleDatabase.__init__c C s t j| jt| jd�� d S )Nr )r �dumpr r r )r r r r �save s zPickleDatabase.savec C s
| j | S )N)r )r �keyr r r � __getitem__ s zPickleDatabase.__getitem__c C s || j |< | j� d S )N)r r )r r �valuer r r � __setitem__ s
zPickleDatabase.__setitem__c C s
t | j�S )N)�strr )r r r r �__repr__" s zPickleDatabase.__repr__c C s
|| j kS )N)r )r r r r r � __contains__% s zPickleDatabase.__contains__N)
r �
__module__� __qualname__�__doc__r r r r r r r r r r r s r )r r � traceback�objectr r r r r �<module> s