#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import ipaddress

import json

import functools

import statistics

from pathlib import Path

from datetime import datetime

import jsonschema

import strict_rfc3339

from flask import current_app

from pkg_resources import resource_filename

from ..utils import check_ip

schema = json.load(

    open(

        resource_filename('ennstatus.api', 'schema/server.json'),

        encoding='utf-8'

    )

)

validate = functools.partial(

    jsonschema.validate,

    schema=schema,

    format_checker=jsonschema.FormatChecker()

)

def calculate_weight(data):

    obj = {}

    for subkey in ('1_week', '1_month', '3_months', '1_year', '5_years'):

        try:

            subdata = data[subkey]

        except KeyError:

            continue

        if values:

            obj[subkey] = statistics.mean(values) * 100

        else:

            obj[subkey] = None

    return obj

class ServerEncoder(json.JSONEncoder):

    def default(self, obj):

        if isinstance(obj, (ipaddress.IPv4Address, ipaddress.IPv6Address)):

            return str(obj)

        if isinstance(obj, datetime):

            return strict_rfc3339.timestamp_to_rfc3339_utcoffset(

                obj.timestamp()

            )

        return json.JSONEncoder.default(self, obj)

            raise e

        for key in ('ip', 'ip6'):

            if key in server:

                address = ipaddress.ip_address(server[key])

                if not check_ip(address):

                    raise ValueError('{} is not accepted!\n'.format(key))

        return True

    def save(self):

        filepath = Path('data') / (self.name.lower() + '.json')

        try:

            with filepath.open(mode='w', encoding='utf-8') as f:

                json.dump(self.__dict__, f, cls=ServerEncoder)

        except Exception as e:

            raise e

    def update_weights(self):

        if self.type not in ('exit', 'relay'):

            raise NotImplementedError

        self.mean_guard_probability = calculate_weight(

        )

        self.mean_middle_probability = calculate_weight(

        )

        self.mean_consensus_weight_fraction = calculate_weight(

        )

    def check_status(self):

        now = datetime.utcnow()

        delta = now - self.last_updated

        if delta.seconds >= 3600:

            self.status = False

        elif delta.seconds >= 600:

            self.status = None

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import ipaddress

import json

from datetime import datetime

from flask import (Blueprint, request, current_app, jsonify, render_template,

                   abort)

from werkzeug.exceptions import BadRequest

import strict_rfc3339

import pygeoip

from ennstatus import csrf

from ennstatus.status.functions import (single_server, all_servers,

                                        all_servers_by_type)

from .model import Server

from .auth import httpauth

api_page = Blueprint('api', __name__)

gi4 = pygeoip.GeoIP('/usr/share/GeoIP/GeoIP.dat', pygeoip.MEMORY_CACHE)

gi6 = pygeoip.GeoIP('/usr/share/GeoIP/GeoIPv6.dat', pygeoip.MEMORY_CACHE)

@csrf.exempt

@api_page.route('/update', methods=('POST',))

@httpauth.login_required

def update():

    current_app.logger.info('Handling update')

    try:

        servers = current_app.config['ENNSTATUS_SERVERS']

    except KeyError as e:

        current_app.logger.error(str(e))

        return abort(500)

                data['ip'] = ip

            elif temp_ip.verison == 6:

                data['ip6'] = ip

    try:

        country = gi4.country_name_by_addr(ip)

    except pygeoip.GeoIPError:

        country = gi6.country_name_by_addr(ip)

    data['country'] = country

    data['last_updated'] = strict_rfc3339.timestamp_to_rfc3339_utcoffset(

        datetime.utcnow().timestamp()

    )

    try:

        server = Server.from_dict(data)

    except Exception as e:

        current_app.logger.warning(' '.join([str(e), str(data)]))

        return str(e), 409, {'Content-Type': 'text/plain'}

    try:

        server.update_weights()

    except NotImplementedError:

        pass

    try:

        server.save()

    except Exception as e:

        current_app.logger.error(str(e))

        return str(e), 500, {'Content-Type': 'text/plain'}

    current_app.logger.info('Return result')

    return (

        server.json(), 201,

        {

            'Location': '/api/export/json/single?server_name={}'.format(

                server.name

            )

        }

    )

@api_page.route('/export', defaults={'server_type': 'all'})

@api_page.route(('/export/<any("all", "exit", "bridge", "relay", "single")'

                 ':server_type>'))

def export(server_type):

    current_app.logger.info('Handling export')

Babel==2.1.1

click==5.1

Flask-Bootstrap==3.3.5.6

Flask-HTTPAuth==2.6.0

Flask-Mail==0.9.1

Flask-Moment==0.5.1

Flask-WTF==0.12

Flask==0.10.1

jsonschema==2.5.1

pygeoip==0.3.2

python-gnupg==0.3.7

strict-rfc3339==0.5

OnionPy